dair-ai/emotion
Viewer • Updated • 437k • 34.2k • 443
How to use ale-dp/distilbert-base-uncased-finetuned-emotion with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="ale-dp/distilbert-base-uncased-finetuned-emotion") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ale-dp/distilbert-base-uncased-finetuned-emotion")
model = AutoModelForSequenceClassification.from_pretrained("ale-dp/distilbert-base-uncased-finetuned-emotion")This model is a fine-tuned variant of distilbert-base-uncased using the emotion dataset. The evaluation results demonstrate its performance:
| Epoch | Training Loss | Validation Loss | Accuracy | F1 |
|---|---|---|---|---|
| 1 | 0.1703 | 0.1709 | 0.9355 | 0.9361 |
| 2 | 0.1115 | 0.1595 | 0.9335 | 0.9335 |
Base model
distilbert/distilbert-base-uncased