distilbert-emotion-classifier

DistilBERT Emotion Classifier

A fine-tuned DistilBERT model for emotion classification on Twitter text data.

Model Description

This model is a fine-tuned version of distilbert-base-uncased on the emotion dataset. It classifies text into 6 emotion categories.

Emotion Labels

Training Details

Training Configuration

• Base Model: distilbert-base-uncased
• Training Epochs: 3
• Learning Rate: 2e-5
• Batch Size: 16
• Weight Decay: 0.01
• FP16: Enabled

Performance

Training Progress

Usage

from transformers import pipeline

classifier = pipeline("text-classification", model="YOUR_USERNAME/distilbert-emotion-classifier")

result = classifier("I am so happy today!")
print(result)
# [{'label': 'joy', 'score': 0.999}]

Or load the model directly:

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("YOUR_USERNAME/distilbert-emotion-classifier")
model = AutoModelForSequenceClassification.from_pretrained("YOUR_USERNAME/distilbert-emotion-classifier")

Limitations

• The model may struggle with sarcasm and irony
• Typos can affect predictions
• The "surprise" category has lower accuracy compared to others

Citation

If you use this model, please cite the original emotion dataset:

@inproceedings{saravia-etal-2018-carer,
    title = "{CARER}: Contextualized Affect Representations for Emotion Recognition",
    author = "Saravia, Elvis  and Liu, Hsien-Chi Toby  and Huang, Yen-Hao  and Wu, Junlin  and Chen, Yi-Shin",
    booktitle = "Proceedings of the 2018 Conference on Empirical Methods in Natural Language Processing",
    year = "2018",
    publisher = "Association for Computational Linguistics",
}