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
108
Safetensors
Model size
67M params
Tensor type
F32
ยท
Inference Providers NEW
This model is not currently available via any of the supported third-party Inference Providers, and the model is not deployed on the HF Inference API.

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

Finetuned
(7430)
this model

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

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