jitesh/emotion-english
jitesh
Clasificación de texto
Modelo de clasificación de texto para identificar emociones en inglés, basado en Transformers y PyTorch. El modelo puede distinguir entre 20 etiquetas de emociones diferentes.
Como usar
Aquí se muestra cómo utilizar este modelo para obtener la etiqueta de emoción de un texto dado:
from transformers import AutoModelForSequenceClassification, pipeline
model_name = 'jitesh/emotion-english'
model = AutoModelForSequenceClassification.from_pretrained(model_name)
classifier = pipeline('text-classification', model=model, tokenizer=model_name)
text = 'I can\'t wait any longer'
prediction = classifier(text)
print(prediction[0], text)
La línea de código anterior produce el siguiente resultado:
{'label': 'impatient', 'score': 0.924211859703064} I can't wait any longer
Funcionalidades
- Clasificación de texto
- Transformers
- PyTorch
- Safetensors
- 20 clases de emoción
Casos de uso
- Clasificación de emociones en texto
- Análisis de sentimientos
- Monitoreo de redes sociales
- Atención al cliente