Training completed!

README.md

@@ -1,32 +1,15 @@
 ---
+library_name: transformers
 license: apache-2.0
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
-datasets:
-- emotion
 metrics:
 - accuracy
 - f1
 model-index:
 - name: distilbert-base-uncased-finetuned-emotion
-  results:
-  - task:
-      name: Text Classification
-      type: text-classification
-    dataset:
-      name: emotion
-      type: emotion
-      config: split
-      split: validation
-      args: split
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.9215
-    - name: F1
-      type: f1
-      value: 0.9214718883562769
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,11 +17,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilbert-base-uncased-finetuned-emotion
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
+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.2186
-- Accuracy: 0.9215
-- F1: 0.9215
+- Loss: 0.2117
+- Accuracy: 0.9275
+- F1: 0.9275
 
 ## Model description
 
@@ -69,13 +52,13 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| 0.8382        | 1.0   | 250  | 0.3176          | 0.905    | 0.9037 |
-| 0.2522        | 2.0   | 500  | 0.2186          | 0.9215   | 0.9215 |
+| 0.8106        | 1.0   | 250  | 0.3030          | 0.908    | 0.9064 |
+| 0.2435        | 2.0   | 500  | 0.2117          | 0.9275   | 0.9275 |
 
 
 ### Framework versions
 
-- Transformers 4.40.2
-- Pytorch 2.2.2+cu121
-- Datasets 2.19.1
+- Transformers 4.44.2
+- Pytorch 2.1.0+cu118
+- Datasets 2.21.0
 - Tokenizers 0.19.1