optimum/distilbert-base-uncased-finetuned-sst-2-english
optimum
Clasificación de texto
Este modelo es un punto de control de ajuste fino de DistilBERT-base-uncased, afinado en SST-2. Este modelo alcanza una precisión de 91.3 en el conjunto de desarrollo (en comparación, la versión Bert bert-base-uncased alcanza una precisión de 92.7). Para más detalles sobre DistilBERT, alentamos a los usuarios a consultar esta tarjeta del modelo.
Como usar
Parámetros de ajuste fino
learning_rate = 1e-5
batch_size = 32
warmup = 600
max_seq_length = 128
num_train_epochs = 3.0
API de inferencia para clasificación de texto
Este modelo aún no tiene suficiente actividad para ser desplegado en la API de inferencia (sin servidor) todavía. Aumenta su visibilidad social y vuelve a verificar más tarde, o despliega a Puntos de entrada para inferencia (dedicados) en su lugar.
Funcionalidades
- Clasificación de texto
- Transformadores
- ONNX
- sst2
- Puntos de entrada para inferencia
Casos de uso
- Clasificación binaria de sentimientos en texto
- Análisis de sentimientos en redes sociales
- Clasificación de reseñas de productos