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.6311
-- Precision: 0.5170
-- Recall: 0.5666
-- F1: 0.5406
-- Accuracy: 0.8126
+- Loss: 0.5661
+- Precision: 0.5387
+- Recall: 0.5925
+- F1: 0.5644
+- Accuracy: 0.8318
 
 ## Model description
 
@@ -54,8 +54,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| 0.6285        | 1.9973 | 1474 | 0.6449          | 0.5123    | 0.5238 | 0.5180 | 0.8125   |
-| 0.5081        | 3.9946 | 2948 | 0.6311          | 0.5170    | 0.5666 | 0.5406 | 0.8126   |
+| 0.6022        | 1.9970 | 1332 | 0.5910          | 0.5313    | 0.5817 | 0.5553 | 0.8284   |
+| 0.4725        | 3.9940 | 2664 | 0.5661          | 0.5387    | 0.5925 | 0.5644 | 0.8318   |
 
 
 ### Framework versions

config.json

@@ -11,108 +11,92 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "B-RELIGION",
-    "1": "B-DATETIME",
-    "2": "B-AGE",
-    "3": "I-QUANTITY",
-    "4": "I-AGE",
-    "5": "I-ROLE",
-    "6": "I-SKILL",
-    "7": "I-RELIGION",
-    "8": "I-TRANSPORTATION",
-    "9": "I-LAW",
-    "10": "I-PHONE",
-    "11": "B-EVENT",
-    "12": "B-QUANTITY",
-    "13": "I-EMAIL",
-    "14": "B-ACTION",
-    "15": "B-ETHNICITY",
-    "16": "I-MISCELLANEOUS",
-    "17": "B-PERSON",
-    "18": "I-PERSON",
-    "19": "I-MONEY",
-    "20": "B-EDUCATION",
-    "21": "B-LOCATION",
-    "22": "I-GENDER",
-    "23": "B-ROLE",
-    "24": "B-MISCELLANEOUS",
-    "25": "I-EVENT",
-    "26": "B-SKILL",
-    "27": "B-EMAIL",
-    "28": "B-MONEY",
-    "29": "I-ACTION",
-    "30": "B-PRODUCT",
-    "31": "B-GENDER",
-    "32": "I-ETHNICITY",
-    "33": "B-ADDRESS",
-    "34": "I-DISEASE",
-    "35": "O",
-    "36": "I-ORGANIZATION",
-    "37": "I-PRODUCT",
-    "38": "B-LAW",
-    "39": "I-DATETIME",
-    "40": "B-PHONE",
-    "41": "I-ADDRESS",
-    "42": "I-LOCATION",
-    "43": "B-DISEASE",
-    "44": "I-EDUCATION",
-    "45": "B-ORGANIZATION",
-    "46": "B-URL",
-    "47": "I-URL",
-    "48": "B-TRANSPORTATION"
+    "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",
+    "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": "B-QUANTITY",
+    "20": "I-AGE",
+    "21": "O",
+    "22": "I-PRODUCT",
+    "23": "B-EVENT",
+    "24": "B-ACTION",
+    "25": "I-ORGANIZATION",
+    "26": "B-PERSON",
+    "27": "I-TRANSPORTATION",
+    "28": "I-EMAIL",
+    "29": "B-AGE",
+    "30": "B-EMAIL",
+    "31": "I-MISCELLANEOUS",
+    "32": "I-LOCATION",
+    "33": "I-DATETIME",
+    "34": "I-QUANTITY",
+    "35": "B-LOCATION",
+    "36": "B-MONEY",
+    "37": "I-MONEY",
+    "38": "B-MISCELLANEOUS",
+    "39": "I-PHONE",
+    "40": "B-LAW"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "B-ACTION": 14,
-    "B-ADDRESS": 33,
-    "B-AGE": 2,
-    "B-DATETIME": 1,
-    "B-DISEASE": 43,
-    "B-EDUCATION": 20,
-    "B-EMAIL": 27,
-    "B-ETHNICITY": 15,
-    "B-EVENT": 11,
-    "B-GENDER": 31,
-    "B-LAW": 38,
-    "B-LOCATION": 21,
-    "B-MISCELLANEOUS": 24,
-    "B-MONEY": 28,
-    "B-ORGANIZATION": 45,
-    "B-PERSON": 17,
-    "B-PHONE": 40,
-    "B-PRODUCT": 30,
-    "B-QUANTITY": 12,
-    "B-RELIGION": 0,
-    "B-ROLE": 23,
-    "B-SKILL": 26,
-    "B-TRANSPORTATION": 48,
-    "B-URL": 46,
-    "I-ACTION": 29,
-    "I-ADDRESS": 41,
-    "I-AGE": 4,
-    "I-DATETIME": 39,
-    "I-DISEASE": 34,
-    "I-EDUCATION": 44,
-    "I-EMAIL": 13,
-    "I-ETHNICITY": 32,
-    "I-EVENT": 25,
-    "I-GENDER": 22,
-    "I-LAW": 9,
-    "I-LOCATION": 42,
-    "I-MISCELLANEOUS": 16,
-    "I-MONEY": 19,
-    "I-ORGANIZATION": 36,
-    "I-PERSON": 18,
-    "I-PHONE": 10,
-    "I-PRODUCT": 37,
-    "I-QUANTITY": 3,
-    "I-RELIGION": 7,
-    "I-ROLE": 5,
-    "I-SKILL": 6,
-    "I-TRANSPORTATION": 8,
-    "I-URL": 47,
-    "O": 35
+    "B-ACTION": 24,
+    "B-AGE": 29,
+    "B-DATETIME": 7,
+    "B-DISEASE": 15,
+    "B-EDUCATION": 10,
+    "B-EMAIL": 30,
+    "B-EVENT": 23,
+    "B-GENDER": 14,
+    "B-LAW": 40,
+    "B-LOCATION": 35,
+    "B-MISCELLANEOUS": 38,
+    "B-MONEY": 36,
+    "B-ORGANIZATION": 0,
+    "B-PERSON": 26,
+    "B-PHONE": 18,
+    "B-PRODUCT": 8,
+    "B-QUANTITY": 19,
+    "B-ROLE": 12,
+    "B-TRANSPORTATION": 13,
+    "B-URL": 11,
+    "I-ACTION": 17,
+    "I-AGE": 20,
+    "I-DATETIME": 33,
+    "I-DISEASE": 6,
+    "I-EDUCATION": 2,
+    "I-EMAIL": 28,
+    "I-EVENT": 4,
+    "I-GENDER": 1,
+    "I-LAW": 5,
+    "I-LOCATION": 32,
+    "I-MISCELLANEOUS": 31,
+    "I-MONEY": 37,
+    "I-ORGANIZATION": 25,
+    "I-PERSON": 3,
+    "I-PHONE": 39,
+    "I-PRODUCT": 22,
+    "I-QUANTITY": 34,
+    "I-ROLE": 9,
+    "I-TRANSPORTATION": 27,
+    "I-URL": 16,
+    "O": 21
   },
   "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:7da698a0b0b73061c3e4dd055ca3bb894d4980e9cc6a1ff3f9e35306b2574b30
-size 532441668
+oid sha256:1540f17794de85514cff467220f8e7497ca8598237842462ae275d147bb45bdc
+size 532417060

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:66c5c84e18537950235026bfa044f04fbfe7bd929860649bc9644691cddba0dc
+oid sha256:55dc49d7d2d9b370c043ac901d2f9865448b220be4e79d9cc1a16f2e44c408ca
 size 5112