Edit model card

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
8
Safetensors
Model size
67M params
Tensor type
F32
·

Finetuned from

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

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