End of training

README.md

@@ -3,26 +3,11 @@ license: apache-2.0
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
-datasets:
-- emotion
 metrics:
 - accuracy
 model-index:
 - name: results
-  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.919
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # results
 
-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.4055
-- Accuracy: 0.919
+- Loss: 0.8835
+- Accuracy: 0.6862
 
 ## Model description
 
@@ -52,22 +37,21 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 8.42e-05
+- learning_rate: 5e-05
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 4
+- num_epochs: 3
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.1735        | 1.0   | 1000 | 0.2957          | 0.919    |
-| 0.1154        | 2.0   | 2000 | 0.2450          | 0.9305   |
-| 0.072         | 3.0   | 3000 | 0.2936          | 0.935    |
-| 0.032         | 4.0   | 4000 | 0.3145          | 0.938    |
+| No log        | 1.0   | 179  | 0.6302          | 0.6440   |
+| No log        | 2.0   | 358  | 0.6395          | 0.6712   |
+| 0.4692        | 3.0   | 537  | 0.8529          | 0.6764   |
 
 
 ### Framework versions

config.json

@@ -8,23 +8,7 @@
   "dim": 768,
   "dropout": 0.1,
   "hidden_dim": 3072,
-  "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4",
-    "5": "LABEL_5"
-  },
   "initializer_range": 0.02,
-  "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2,
-    "LABEL_3": 3,
-    "LABEL_4": 4,
-    "LABEL_5": 5
-  },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7d54e4c4738766f78da46d6cd0dded2a4577f0ccaf3b33bb75a896fb44324668
-size 267844872
+oid sha256:f69324d10cdc73964c32464f405ad0069c508e9f932626911a28d379bbc8d2e0
+size 267832560

runs/Mar11_04-37-43_f6b7ca2de819/events.out.tfevents.1710131879.f6b7ca2de819.11996.1

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9cd2c7f902c1e9e2a0bdcad2181ef822942d01b002e7261692a5283292f2d6bd
-size 5637
+oid sha256:8a32d85dd8e37227b8d840247e3dededafd98877a17e17bb800ee6778e5ae467
+size 6314

runs/Mar11_04-37-43_f6b7ca2de819/events.out.tfevents.1710132838.f6b7ca2de819.11996.3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:26802624fd55076875ec5260dfb55e9e78d723adff9cfcb7da0275871b9f0d56
+size 411

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0acfda34f5c210b7a98ed255121422bfae7eebc6c0740b69b494a866d43d0b24
+oid sha256:370f97111278dc5bae72fbbf275aed53cb61ea6388e34a2a26d1d6ae9fb34780
 size 4920