End of training

README.md

@@ -20,11 +20,11 @@ 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: 0.0004
-- Precision: 1.0
-- Recall: 1.0
-- F1: 1.0
-- Accuracy: 1.0
+- Loss: 0.0914
+- Precision: 0.9662
+- Recall: 0.9470
+- F1: 0.9565
+- Accuracy: 0.9953
 
 ## Model description
 
@@ -49,17 +49,15 @@ 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: 5
+- num_epochs: 3
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1  | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:---:|:--------:|
-| No log        | 1.0   | 50   | 0.0015          | 1.0       | 1.0    | 1.0 | 1.0      |
-| No log        | 2.0   | 100  | 0.0007          | 1.0       | 1.0    | 1.0 | 1.0      |
-| No log        | 3.0   | 150  | 0.0005          | 1.0       | 1.0    | 1.0 | 1.0      |
-| No log        | 4.0   | 200  | 0.0004          | 1.0       | 1.0    | 1.0 | 1.0      |
-| No log        | 5.0   | 250  | 0.0004          | 1.0       | 1.0    | 1.0 | 1.0      |
+| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
+| No log        | 1.0   | 10   | 0.4124          | 0.5690    | 0.4371 | 0.4944 | 0.8733   |
+| No log        | 2.0   | 20   | 0.1510          | 0.9236    | 0.8808 | 0.9017 | 0.9884   |
+| No log        | 3.0   | 30   | 0.0914          | 0.9662    | 0.9470 | 0.9565 | 0.9953   |
 
 
 ### Framework versions

config.json

@@ -9,15 +9,15 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "DESC",
-    "1": "DATE",
-    "2": "CURRENCY"
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "CURRENCY": 2,
-    "DATE": 1,
-    "DESC": 0
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_2": 2
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:421320d11055aa33ae2e5b5209b6e376f7201ce2c6915dc0fe77b82089daf22a
+oid sha256:60bcb4e32cf09ae44a3ce0501ce28790068a2e395019768ef06092eb12213d55
 size 265473092

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d49889f978231be49eea47c4a3d0a8f9198bd524af878ced4d9b3bd2b0e51a8b
+oid sha256:70e88d0ecafe200053f86690f2ea148bc41f80d68b50f02b96e85a85fdc854bb
 size 5112