End of training

README.md

@@ -17,8 +17,8 @@ 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 the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.0126
-- Accuracy: 0.25
+- Loss: 2.5873
+- Accuracy: 0.1544
 
 ## Model description
 
@@ -43,18 +43,27 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 2.0
+- num_epochs: 2.5
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 2.1499        | 0.29  | 60   | 2.1015          | 0.1991   |
-| 2.105         | 0.58  | 120  | 2.0680          | 0.2158   |
-| 2.0741        | 0.87  | 180  | 2.0512          | 0.2384   |
-| 2.0064        | 1.16  | 240  | 2.0333          | 0.2398   |
-| 1.903         | 1.45  | 300  | 2.0206          | 0.2384   |
-| 1.9054        | 1.74  | 360  | 2.0126          | 0.25     |
+| 2.6964        | 0.16  | 60   | 2.6756          | 0.0834   |
+| 2.6947        | 0.31  | 120  | 2.6670          | 0.0947   |
+| 2.6716        | 0.47  | 180  | 2.6428          | 0.1139   |
+| 2.6315        | 0.63  | 240  | 2.6252          | 0.1268   |
+| 2.6113        | 0.79  | 300  | 2.6205          | 0.1310   |
+| 2.5936        | 0.94  | 360  | 2.6137          | 0.1293   |
+| 2.5298        | 1.1   | 420  | 2.6065          | 0.1381   |
+| 2.4691        | 1.26  | 480  | 2.6097          | 0.1452   |
+| 2.4645        | 1.42  | 540  | 2.6015          | 0.1519   |
+| 2.4404        | 1.57  | 600  | 2.5912          | 0.1414   |
+| 2.4435        | 1.73  | 660  | 2.5890          | 0.1464   |
+| 2.4425        | 1.89  | 720  | 2.5928          | 0.1514   |
+| 2.4038        | 2.05  | 780  | 2.5880          | 0.1481   |
+| 2.2601        | 2.2   | 840  | 2.5941          | 0.1423   |
+| 2.2858        | 2.36  | 900  | 2.5873          | 0.1544   |
 
 
 ### Framework versions

config.json

@@ -11,25 +11,37 @@
   "id2label": {
     "0": "Andy",
     "1": "Angela",
-    "2": "Dwight",
-    "3": "Erin",
-    "4": "Jim",
-    "5": "Kevin",
-    "6": "Michael",
-    "7": "Oscar",
-    "8": "Pam"
+    "2": "Darryl",
+    "3": "Dwight",
+    "4": "Erin",
+    "5": "Jan",
+    "6": "Jim",
+    "7": "Kelly",
+    "8": "Kevin",
+    "9": "Michael",
+    "10": "Oscar",
+    "11": "Pam",
+    "12": "Phyllis",
+    "13": "Ryan",
+    "14": "Toby"
   },
   "initializer_range": 0.02,
   "label2id": {
     "Andy": 0,
     "Angela": 1,
-    "Dwight": 2,
-    "Erin": 3,
-    "Jim": 4,
-    "Kevin": 5,
-    "Michael": 6,
-    "Oscar": 7,
-    "Pam": 8
+    "Darryl": 2,
+    "Dwight": 3,
+    "Erin": 4,
+    "Jan": 5,
+    "Jim": 6,
+    "Kelly": 7,
+    "Kevin": 8,
+    "Michael": 9,
+    "Oscar": 10,
+    "Pam": 11,
+    "Phyllis": 12,
+    "Ryan": 13,
+    "Toby": 14
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7e9ac1601dd80fc561ac69652ea39e46cd850a9fbe5f8184f70b3fac630cdcbe
-size 267875629
+oid sha256:4f6e7388bce18b2b605553aea45357ee7a822d9f49d16d49e39eb9bbaa29bade
+size 267894125

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8f231faf32b9da9e19234193c5053fa3f27340634ef2afad0312a45e574fbf2d
+oid sha256:a6485da611124e3d8ce3a90f9e543316e68cc53c2a3f561390488a73fcbbd490
 size 4027