distilbert-finetuned-emotion

DistilBERT Fine-Tuned for Emotion Detection

Model Description

This model is a fine-tuned version of distilbert-base-uncased on the Emotion dataset for detecting emotions in English Twitter text.

The model classifies text into six basic emotions:

Intended Use

• Detecting emotional states in short English text (tweets, messages, comments)
• Social media monitoring and sentiment analysis
• Customer feedback emotion classification

Training Details

Evaluation Results

• Accuracy: ~92%
• F1-Score (weighted): ~92%

Usage

from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="<your-username>/distilbert-finetuned-emotion"
)

result = classifier("I am feeling great today!")
print(result)
# [{'label': 'joy', 'score': 0.98}]

Limitations

• Trained on Twitter data — may not generalize well to formal text
• Class imbalance: joy and sadness are overrepresented
• surprise and love are the most challenging categories
• Only 1 epoch of fine-tuning — additional training may improve performance

License

Apache 2.0