Views
No views yet
j-hartmann/emotion-english-distilroberta-base on the dair-ai/emotion dataset.| Metric | Baseline | Fine-Tuned |
|---|---|---|
| Macro F1 | 0.869 | 0.892 |
| Accuracy | 0.914 | 0.930 |
1from transformers import pipeline
2classifier = pipeline("text-classification", model="YDVJIYA/distilroberta-base-finetuned-emotion")
3result = classifier("I feel hopeless and tired today")
4print(result)
5
6Training Configuration
7Epochs: 3
8Batch size: 32
9Learning rate: 2e-5
10Warmup ratio: 0.1
11Weight decay: 0.01
12Mixed precision (fp16)
13Seed: 42
14
15Author
16Trained by Jiya Yadav as part of an end-to-end agentic AI platform.