Downloads · 30 days
12
4% of all-time downloads
distilroberta-base-finetuned-emotion
YDVJIYA/distilroberta-base-finetuned-emotion
distilroberta-base-finetuned-emotion is a machine learning model from YDVJIYA. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Fine-tuned from j-hartmann/emotion-english-distilroberta-base on the dair-ai/emotion dataset.
All-time downloads
318
Public
Parameters
82.1M
329 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors329 MB · 98%
At a glance
- License
- mit
- Model type
- roberta
- Access
- Public
- Created
- Jun 30, 2026
- Updated
- Jul 2, 2026
- SHA
- d09c0382
Base models
Datasets
Model card
From the Hugging Face model README
DistilRoBERTa Fine-Tuned for Emotion Classification
Fine-tuned from j-hartmann/emotion-english-distilroberta-base on the dair-ai/emotion dataset.
Part of the Mental Health Agentic AI Platform.
Performance
| Metric | Baseline | Fine-Tuned |
|---|---|---|
| Macro F1 | 0.869 | 0.892 |
| Accuracy | 0.914 | 0.930 |
Labels
- 0: sadness
- 1: joy
- 2: love
- 3: anger
- 4: fear
- 5: surprise
Usage
from transformers import pipeline
classifier = pipeline("text-classification", model="YDVJIYA/distilroberta-base-finetuned-emotion")
result = classifier("I feel hopeless and tired today")
print(result)
Training Configuration
Epochs: 3
Batch size: 32
Learning rate: 2e-5
Warmup ratio: 0.1
Weight decay: 0.01
Mixed precision (fp16)
Seed: 42
Author
Trained by Jiya Yadav as part of an end-to-end agentic AI platform.
- Type
- roberta
- License
- mit
- Languages
- en
- Created
- Jun 30, 2026
- Updated
- Jul 2, 2026