YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
π§ DistilBERT Emotion Classification (by Subhra Sikha Sahoo)
This model is a fine-tuned version of distilbert-base-uncased for emotion classification.
It predicts the emotion expressed in a given text such as joy, sadness, anger, fear, love, surprise, etc.
π Model Details
- Base Model: distilbert-base-uncased
- Trained By: Subhra Sikha Sahoo
- Task: Multi-label Emotion Classification
- Language: English
- Dataset Size: 20,000 samples
π Usage Example
from transformers import pipeline
pipe = pipeline("text-classification", model="Subhra-123/distilbert-base-emotion", return_all_scores=True)
text = "I love my friends!"
pred = pipe(text)[0]
emotion = max(pred, key=lambda x: x['score'])
print(f"{text} β {emotion['label']} ({emotion['score']:.2f})")
- Downloads last month
- 5
Inference Providers NEW
This model isn't deployed by any Inference Provider. π Ask for provider support