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.4817
-- Accuracy: 0.1787
+- Loss: 2.0126
+- Accuracy: 0.25
 
 ## Model description
 
@@ -43,27 +43,18 @@ 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: 3.0
+- num_epochs: 2.0
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 2.5596        | 0.2   | 60   | 2.5272          | 0.1201   |
-| 2.5284        | 0.4   | 120  | 2.5153          | 0.1206   |
-| 2.4807        | 0.6   | 180  | 2.4858          | 0.1449   |
-| 2.4921        | 0.8   | 240  | 2.5119          | 0.1201   |
-| 2.4656        | 1.0   | 300  | 2.4461          | 0.1636   |
-| 2.3184        | 1.2   | 360  | 2.4390          | 0.1671   |
-| 2.2852        | 1.4   | 420  | 2.4665          | 0.1575   |
-| 2.2561        | 1.59  | 480  | 2.4464          | 0.1731   |
-| 2.2903        | 1.79  | 540  | 2.4487          | 0.1752   |
-| 2.242         | 1.99  | 600  | 2.4399          | 0.1731   |
-| 2.0247        | 2.19  | 660  | 2.4595          | 0.1777   |
-| 2.0127        | 2.39  | 720  | 2.4709          | 0.1731   |
-| 1.9785        | 2.59  | 780  | 2.4754          | 0.1752   |
-| 2.0006        | 2.79  | 840  | 2.4717          | 0.1843   |
-| 1.9937        | 2.99  | 900  | 2.4817          | 0.1787   |
+| 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     |
 
 
 ### Framework versions

config.json

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

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:32cb8d7061b30af8890be3dd3a487d3801654d03d18a26632bbeff67b8ef0eca
-size 267887917
+oid sha256:7e9ac1601dd80fc561ac69652ea39e46cd850a9fbe5f8184f70b3fac630cdcbe
+size 267875629

training_args.bin

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