jungnerd
/

distilbert-base-uncased-finetuned-emotion

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

jungnerd commited on Nov 5, 2024

Commit

0e43264

·

verified ·

1 Parent(s): 89432ee

Training completed!

Files changed (1) hide show

README.md +5 -5

README.md CHANGED Viewed

@@ -19,9 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2212
-- Accuracy: 0.922
-- F1: 0.9219
 ## Model description
@@ -52,8 +52,8 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| 0.8495        | 1.0   | 250  | 0.3211          | 0.9025   | 0.9012 |
-| 0.2526        | 2.0   | 500  | 0.2212          | 0.922    | 0.9219 |
 ### Framework versions

 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.2249
+- Accuracy: 0.923
+- F1: 0.9231
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
+| 0.8381        | 1.0   | 250  | 0.3319          | 0.9055   | 0.9050 |
+| 0.2536        | 2.0   | 500  | 0.2249          | 0.923    | 0.9231 |
 ### Framework versions