distilbert-imdb
lvwerra
Clasificación de texto
Este modelo es una versión ajustada de distilbert-base-uncased en el conjunto de datos imdb. Se logra los siguientes resultados en el conjunto de evaluación: Pérdida: 0.1903, Precisión: 0.928
Como usar
Este modelo es ideal para tareas de clasificación de texto en el conjunto de datos imdb.
from transformers import pipeline
classifier = pipeline('sentiment-analysis', model='lvwerra/distilbert-imdb')
result = classifier('Este es el mejor curso que he tomado!')
print(result)
Funcionalidades
- Clasificación de texto
- Transformadores
- PyTorch
- TensorBoard
- Generado a partir de Trainer Eval Results
- Pérdida de entrenamiento: 0.1903
- Precisión: 0.928
Casos de uso
- Clasificación de sentimientos en reseñas de películas
- Análisis de opiniones de usuarios
- Filtrado de contenido en redes sociales