End of training

README.md

@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1902
-- Precision: 0.9519
-- Recall: 0.9518
-- F1: 0.9518
-- Accuracy: 0.9478
+- Loss: 0.1677
+- Precision: 0.9005
+- Recall: 0.9037
+- F1: 0.9021
+- Accuracy: 0.9532
 
 ## Model description
 
@@ -56,9 +56,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| No log        | 1.0   | 243  | 0.2151          | 0.9446    | 0.9467 | 0.9456 | 0.9418   |
-| No log        | 2.0   | 486  | 0.1930          | 0.9496    | 0.9504 | 0.9500 | 0.9460   |
-| 0.3146        | 3.0   | 729  | 0.1902          | 0.9519    | 0.9518 | 0.9518 | 0.9478   |
+| No log        | 1.0   | 451  | 0.1872          | 0.8908    | 0.8907 | 0.8908 | 0.9472   |
+| 0.3208        | 2.0   | 902  | 0.1698          | 0.9005    | 0.9012 | 0.9009 | 0.9527   |
+| 0.1569        | 3.0   | 1353 | 0.1677          | 0.9005    | 0.9037 | 0.9021 | 0.9532   |
 
 
 ### Framework versions

config.json

@@ -9,35 +9,35 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "I-VIP",
-    "1": "B-MED",
-    "2": "B-ORG",
-    "3": "I-PER",
-    "4": "I-LEG",
-    "5": "O",
-    "6": "B-LOC",
-    "7": "I-ORG",
-    "8": "B-LEG",
-    "9": "B-VIP",
-    "10": "B-PER",
-    "11": "I-MED",
-    "12": "I-LOC"
+    "0": "B-LOC",
+    "1": "I-LEG",
+    "2": "B-LEG",
+    "3": "I-ORG",
+    "4": "B-ORG",
+    "5": "B-VIP",
+    "6": "I-VIP",
+    "7": "B-PER",
+    "8": "I-PER",
+    "9": "O",
+    "10": "I-LOC",
+    "11": "B-MED",
+    "12": "I-MED"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "B-LEG": 8,
-    "B-LOC": 6,
-    "B-MED": 1,
-    "B-ORG": 2,
-    "B-PER": 10,
-    "B-VIP": 9,
-    "I-LEG": 4,
-    "I-LOC": 12,
-    "I-MED": 11,
-    "I-ORG": 7,
-    "I-PER": 3,
-    "I-VIP": 0,
-    "O": 5
+    "B-LEG": 2,
+    "B-LOC": 0,
+    "B-MED": 11,
+    "B-ORG": 4,
+    "B-PER": 7,
+    "B-VIP": 5,
+    "I-LEG": 1,
+    "I-LOC": 10,
+    "I-MED": 12,
+    "I-ORG": 3,
+    "I-PER": 8,
+    "I-VIP": 6,
+    "O": 9
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:716d3468774023ee01a697228449368a75b0d1918f3305cd4e802f727449c263
+oid sha256:6b7f106435a9ae3480d92e9b9172c4607a37809ea3af42117e2ea41041978980
 size 260815972

runs/Nov20_17-11-36_fd34eb50cdc0/events.out.tfevents.1732122748.fd34eb50cdc0.30.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:91b326cb01790c0b2c43514beadbb5d5c6201e305be1e825a12dea10a917c084
+size 7588

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:54f2fb5322aff3a2268498656ed3fa75c600329068a4701240968a1504a64b89
+oid sha256:f6b81b940abbb999399667aa47a997dd488e0ceae04df2e1821b05e41c4b5e8b
 size 5240