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:
- Loss: 0.1595
- Accuracy: 93.35%
- F1 Score: 93.35%
Hyperparameters
- learning_rate: 2e-05
- train_batch_size: 64
- eval_batch_size: 64
- seed: 42
- lr_scheduler_type: linear
- num_epochs: 2
Training results
| 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 |
- Downloads last month
- 4
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.
Model tree for ale-dp/distilbert-base-uncased-finetuned-emotion
Base model
distilbert/distilbert-base-uncased