Downloads · 30 days
5
12% of all-time downloads
zhubaihe1/distilbert-finetuned-emotion
distilbert-finetuned-emotion is a text classification model from zhubaihe1. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
A DistilBERT model fine-tuned for 6-class emotion classification on Twitter data.
Downloads · 30 days
5
12% of all-time downloads
All-time downloads
43
Public
Repo size
—
Likes
0
Public
Click a slice to open those files.
.json467 KB · 67%
From the Hugging Face model README
A DistilBERT model fine-tuned for 6-class emotion classification on Twitter data.
0: sadness, 1: joy, 2: love, 3: anger, 4: fear, 5: surprise
from transformers import pipeline
classifier = pipeline("text-classification", model="your-username/distilbert-finetuned-emotion")
result = classifier("I am so happy today!")
print(result)