Training completed!

README.md

@@ -0,0 +1,81 @@
+---
+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.922
+    - name: F1
+      type: f1
+      value: 0.9219196681926928
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+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.
+It achieves the following results on the evaluation set:
+- Loss: 0.2162
+- Accuracy: 0.922
+- F1: 0.9219
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 2e-05
+- train_batch_size: 64
+- eval_batch_size: 64
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 2
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
+| 0.8226        | 1.0   | 250  | 0.3270          | 0.9035   | 0.9024 |
+| 0.2533        | 2.0   | 500  | 0.2162          | 0.922    | 0.9219 |
+
+
+### Framework versions
+
+- Transformers 4.37.2
+- Pytorch 2.2.1+cu121
+- Datasets 2.17.0
+- Tokenizers 0.15.1

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f69bb94f5679e22f8e88404f97a740e2ce37469d1d53c77b5f363f747f022b6a
+oid sha256:abda52c12383258863e78b0e5a36987639ae2e055704fd6abc9b9cc6963efa43
 size 267844872

trainer_state.json

@@ -11,43 +11,43 @@
     {
       "epoch": 1.0,
       "learning_rate": 1e-05,
-      "loss": 0.8202,
+      "loss": 0.8226,
       "step": 250
     },
     {
       "epoch": 1.0,
-      "eval_accuracy": 0.901,
-      "eval_f1": 0.9000590964742174,
-      "eval_loss": 0.3247230350971222,
-      "eval_runtime": 4.5139,
-      "eval_samples_per_second": 443.079,
-      "eval_steps_per_second": 7.089,
+      "eval_accuracy": 0.9035,
+      "eval_f1": 0.9023761650285413,
+      "eval_loss": 0.326994925737381,
+      "eval_runtime": 4.5192,
+      "eval_samples_per_second": 442.555,
+      "eval_steps_per_second": 7.081,
       "step": 250
     },
     {
       "epoch": 2.0,
       "learning_rate": 0.0,
-      "loss": 0.2547,
+      "loss": 0.2533,
       "step": 500
     },
     {
       "epoch": 2.0,
-      "eval_accuracy": 0.9205,
-      "eval_f1": 0.9204480932224948,
-      "eval_loss": 0.21922191977500916,
-      "eval_runtime": 4.5051,
-      "eval_samples_per_second": 443.938,
-      "eval_steps_per_second": 7.103,
+      "eval_accuracy": 0.922,
+      "eval_f1": 0.9219196681926928,
+      "eval_loss": 0.21616625785827637,
+      "eval_runtime": 4.4445,
+      "eval_samples_per_second": 449.996,
+      "eval_steps_per_second": 7.2,
       "step": 500
     },
     {
       "epoch": 2.0,
       "step": 500,
       "total_flos": 720342861696000.0,
-      "train_loss": 0.5374606018066406,
-      "train_runtime": 260.6287,
-      "train_samples_per_second": 122.78,
-      "train_steps_per_second": 1.918
+      "train_loss": 0.5379159927368165,
+      "train_runtime": 263.9893,
+      "train_samples_per_second": 121.217,
+      "train_steps_per_second": 1.894
     }
   ],
   "logging_steps": 250,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:98ab995a6f6b3efbd34a24c2da3e03ebd6d386366b4c58a3f9f5eac9ece73f5a
+oid sha256:91a91b9edad39c6efaf98433b769dcfef6b569ac32ebfca2ad691c24a3333fb9
 size 4728