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
121
Safetensors
Model size
67M params
Tensor type
F32
ยท
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.

Model tree for ale-dp/distilbert-base-uncased-finetuned-emotion

Finetuned
(7711)
this model

Dataset used to train ale-dp/distilbert-base-uncased-finetuned-emotion

Space using ale-dp/distilbert-base-uncased-finetuned-emotion 1