Downloads · 30 days
8
24% of all-time downloads
Mary154/distilbert-emotion-classifier
distilbert-emotion-classifier is a text classification model from Mary154. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
Fine-tuned DistilBERT model for six-way Twitter emotion classification: sadness, joy, love, anger, fear, and surprise.
Downloads · 30 days
8
24% of all-time downloads
All-time downloads
33
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors268 MB · 100%
From the Hugging Face model README
Fine-tuned DistilBERT model for six-way Twitter emotion classification: sadness, joy, love, anger, fear, and surprise.
Validation metrics:
Test metrics:
Training setup: 3 epochs, learning rate 2e-5, batch size 64, weight decay 0.01, fp16 training, and best-checkpoint loading by validation weighted F1.