End of training

README.md

@@ -19,11 +19,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [NlpHUST/ner-vietnamese-electra-base](https://huggingface.co/NlpHUST/ner-vietnamese-electra-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5321
-- Precision: 0.5840
-- Recall: 0.6268
-- F1: 0.6047
-- Accuracy: 0.8423
+- Loss: 0.5013
+- Precision: 0.5533
+- Recall: 0.6351
+- F1: 0.5914
+- Accuracy: 0.8419
 
 ## Model description
 
@@ -54,8 +54,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| 0.4577        | 1.9981 | 2088 | 0.5131          | 0.5846    | 0.6146 | 0.5992 | 0.8432   |
-| 0.3523        | 3.9962 | 4176 | 0.5321          | 0.5840    | 0.6268 | 0.6047 | 0.8423   |
+| 0.4831        | 1.9990 | 2099 | 0.4981          | 0.5509    | 0.6224 | 0.5845 | 0.8411   |
+| 0.391         | 3.9981 | 4198 | 0.5013          | 0.5533    | 0.6351 | 0.5914 | 0.8419   |
 
 
 ### Framework versions

config.json

@@ -11,84 +11,84 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "B-ORGANIZATION",
-    "1": "I-GENDER",
-    "2": "I-EDUCATION",
-    "3": "I-PERSON",
-    "4": "I-EVENT",
-    "5": "I-LAW",
-    "6": "I-DISEASE",
-    "7": "B-DATETIME",
-    "8": "B-PRODUCT",
-    "9": "I-ROLE",
-    "10": "B-EDUCATION",
+    "0": "B-PHONE",
+    "1": "I-LAW",
+    "2": "B-ACTION",
+    "3": "I-LOCATION",
+    "4": "I-PERSON",
+    "5": "I-URL",
+    "6": "I-TRANSPORTATION",
+    "7": "I-EMAIL",
+    "8": "I-ACTION",
+    "9": "I-DISEASE",
+    "10": "I-QUANTITY-NUMBER",
     "11": "B-URL",
-    "12": "B-ROLE",
-    "13": "B-TRANSPORTATION",
-    "14": "B-GENDER",
-    "15": "B-DISEASE",
-    "16": "I-URL",
-    "17": "I-ACTION",
-    "18": "B-PHONE",
-    "19": "I-AGE",
-    "20": "O",
-    "21": "I-PRODUCT",
-    "22": "B-EVENT",
-    "23": "B-ACTION",
-    "24": "I-ORGANIZATION",
-    "25": "B-PERSON",
-    "26": "B-AGE",
-    "27": "I-TRANSPORTATION",
-    "28": "I-EMAIL",
-    "29": "B-EMAIL",
-    "30": "B-QUANTITY-NUMBER",
-    "31": "I-LOCATION",
-    "32": "I-DATETIME",
-    "33": "B-LOCATION",
-    "34": "I-QUANTITY-NUMBER",
-    "35": "I-PHONE",
-    "36": "B-LAW"
+    "12": "I-GENDER",
+    "13": "B-GENDER",
+    "14": "B-PERSON",
+    "15": "B-LAW",
+    "16": "I-AGE",
+    "17": "I-ROLE",
+    "18": "B-EMAIL",
+    "19": "B-EVENT",
+    "20": "B-PRODUCT",
+    "21": "I-PHONE",
+    "22": "B-ORGANIZATION",
+    "23": "B-QUANTITY-NUMBER",
+    "24": "B-LOCATION",
+    "25": "I-EVENT",
+    "26": "B-EDUCATION",
+    "27": "B-DATETIME",
+    "28": "B-ROLE",
+    "29": "B-DISEASE",
+    "30": "O",
+    "31": "B-AGE",
+    "32": "I-PRODUCT",
+    "33": "I-EDUCATION",
+    "34": "I-DATETIME",
+    "35": "B-TRANSPORTATION",
+    "36": "I-ORGANIZATION"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "B-ACTION": 23,
-    "B-AGE": 26,
-    "B-DATETIME": 7,
-    "B-DISEASE": 15,
-    "B-EDUCATION": 10,
-    "B-EMAIL": 29,
-    "B-EVENT": 22,
-    "B-GENDER": 14,
-    "B-LAW": 36,
-    "B-LOCATION": 33,
-    "B-ORGANIZATION": 0,
-    "B-PERSON": 25,
-    "B-PHONE": 18,
-    "B-PRODUCT": 8,
-    "B-QUANTITY-NUMBER": 30,
-    "B-ROLE": 12,
-    "B-TRANSPORTATION": 13,
+    "B-ACTION": 2,
+    "B-AGE": 31,
+    "B-DATETIME": 27,
+    "B-DISEASE": 29,
+    "B-EDUCATION": 26,
+    "B-EMAIL": 18,
+    "B-EVENT": 19,
+    "B-GENDER": 13,
+    "B-LAW": 15,
+    "B-LOCATION": 24,
+    "B-ORGANIZATION": 22,
+    "B-PERSON": 14,
+    "B-PHONE": 0,
+    "B-PRODUCT": 20,
+    "B-QUANTITY-NUMBER": 23,
+    "B-ROLE": 28,
+    "B-TRANSPORTATION": 35,
     "B-URL": 11,
-    "I-ACTION": 17,
-    "I-AGE": 19,
-    "I-DATETIME": 32,
-    "I-DISEASE": 6,
-    "I-EDUCATION": 2,
-    "I-EMAIL": 28,
-    "I-EVENT": 4,
-    "I-GENDER": 1,
-    "I-LAW": 5,
-    "I-LOCATION": 31,
-    "I-ORGANIZATION": 24,
-    "I-PERSON": 3,
-    "I-PHONE": 35,
-    "I-PRODUCT": 21,
-    "I-QUANTITY-NUMBER": 34,
-    "I-ROLE": 9,
-    "I-TRANSPORTATION": 27,
-    "I-URL": 16,
-    "O": 20
+    "I-ACTION": 8,
+    "I-AGE": 16,
+    "I-DATETIME": 34,
+    "I-DISEASE": 9,
+    "I-EDUCATION": 33,
+    "I-EMAIL": 7,
+    "I-EVENT": 25,
+    "I-GENDER": 12,
+    "I-LAW": 1,
+    "I-LOCATION": 3,
+    "I-ORGANIZATION": 36,
+    "I-PERSON": 4,
+    "I-PHONE": 21,
+    "I-PRODUCT": 32,
+    "I-QUANTITY-NUMBER": 10,
+    "I-ROLE": 17,
+    "I-TRANSPORTATION": 6,
+    "I-URL": 5,
+    "O": 30
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f39c5b408664785f3d03989591503f1ccf15eda8d6a5fd6fcdcd0543f2b4117e
+oid sha256:348f1bf85294f5ed89f197666e691f194f219fa5c706a879c31994c4f11e6e33
 size 532404756

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:16e513e54cbb50baa893755d2671e1b0e27f83c73dbc9d544aa57d433cabaf9b
+oid sha256:161e344d16e5d2258192e23bbd303fc4dd6aa7d5f514c893fba30dbd8f399f6d
 size 5112