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.0914
-- Precision: 0.9662
-- Recall: 0.9470
-- F1: 0.9565
-- Accuracy: 0.9953
+- Loss: 0.0006
+- Precision: 1.0
+- Recall: 1.0
+- F1: 1.0
+- Accuracy: 1.0
 
 ## Model description
 
@@ -53,11 +53,11 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| 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   |
+| 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.0006          | 1.0       | 1.0    | 1.0 | 1.0      |
 
 
 ### Framework versions

config.json

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

model.safetensors

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

training_args.bin

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