michellejieli/emotion_text_classifier
michellejieli
Clasificación de texto
DistilRoBERTa-base es un modelo de transformador que realiza análisis de sentimientos. Fine-tuné el modelo en transcripciones del programa Friends con el objetivo de clasificar emociones a partir de datos textuales, específicamente diálogos de programas o películas de Netflix. El modelo predice 6 emociones de Ekman y una clase neutral. Estas emociones incluyen ira, disgusto, miedo, alegría, neutralidad, tristeza y sorpresa. El modelo es una versión afinada de Emotion English DistilRoBERTa-base y DistilRoBERTa-base. Este modelo se entrenó inicialmente en la siguiente tabla de Emotion English DistilRoBERTa-base:
Como usar
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="michellejieli/emotion_text_classifier")
classifier("I love this!")
Output:
[
{'label': 'joy', 'score': 0.9887555241584778}
]
Funcionalidades
- Clasificación de emociones
- Análisis de sentimientos
- Modelo Fine-tuned
- Compatibilidad con sesiones de entrenamiento y endpoints de inferencia
- Utiliza DistilRoBERTa-base
Casos de uso
- Clasificación de emociones en diálogos de películas o series
- Análisis de sentimientos en texto de redes sociales como Twitter y Reddit
- Investigación y análisis en estudios lingüísticos y psicológicos