lilyyellow
commited on
Commit
•
f86741f
1
Parent(s):
9beda29
End of training
Browse files- README.md +7 -7
- config.json +33 -41
- model.safetensors +2 -2
- training_args.bin +1 -1
README.md
CHANGED
@@ -19,11 +19,11 @@ should probably proofread and complete it, then remove this comment. -->
|
|
19 |
|
20 |
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.
|
21 |
It achieves the following results on the evaluation set:
|
22 |
-
- Loss: 0.
|
23 |
-
- Precision: 0.
|
24 |
-
- Recall: 0.
|
25 |
-
- F1: 0.
|
26 |
-
- Accuracy: 0.
|
27 |
|
28 |
## Model description
|
29 |
|
@@ -54,8 +54,8 @@ The following hyperparameters were used during training:
|
|
54 |
|
55 |
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|
56 |
|:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
|
57 |
-
| 0.
|
58 |
-
| 0.
|
59 |
|
60 |
|
61 |
### Framework versions
|
|
|
19 |
|
20 |
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.
|
21 |
It achieves the following results on the evaluation set:
|
22 |
+
- Loss: 0.5321
|
23 |
+
- Precision: 0.5840
|
24 |
+
- Recall: 0.6268
|
25 |
+
- F1: 0.6047
|
26 |
+
- Accuracy: 0.8423
|
27 |
|
28 |
## Model description
|
29 |
|
|
|
54 |
|
55 |
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|
56 |
|:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
|
57 |
+
| 0.4577 | 1.9981 | 2088 | 0.5131 | 0.5846 | 0.6146 | 0.5992 | 0.8432 |
|
58 |
+
| 0.3523 | 3.9962 | 4176 | 0.5321 | 0.5840 | 0.6268 | 0.6047 | 0.8423 |
|
59 |
|
60 |
|
61 |
### Framework versions
|
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"ElectraForTokenClassification"
|
5 |
],
|
@@ -30,73 +30,65 @@
|
|
30 |
"16": "I-URL",
|
31 |
"17": "I-ACTION",
|
32 |
"18": "B-PHONE",
|
33 |
-
"19": "
|
34 |
-
"20": "
|
35 |
-
"21": "
|
36 |
-
"22": "
|
37 |
-
"23": "B-
|
38 |
-
"24": "
|
39 |
-
"25": "
|
40 |
-
"26": "B-
|
41 |
"27": "I-TRANSPORTATION",
|
42 |
"28": "I-EMAIL",
|
43 |
-
"29": "B-
|
44 |
-
"30": "B-
|
45 |
-
"31": "I-
|
46 |
-
"32": "I-
|
47 |
-
"33": "
|
48 |
-
"34": "I-QUANTITY",
|
49 |
-
"35": "
|
50 |
-
"36": "B-
|
51 |
-
"37": "I-MONEY",
|
52 |
-
"38": "B-MISCELLANEOUS",
|
53 |
-
"39": "I-PHONE",
|
54 |
-
"40": "B-LAW"
|
55 |
},
|
56 |
"initializer_range": 0.02,
|
57 |
"intermediate_size": 3072,
|
58 |
"label2id": {
|
59 |
-
"B-ACTION":
|
60 |
-
"B-AGE":
|
61 |
"B-DATETIME": 7,
|
62 |
"B-DISEASE": 15,
|
63 |
"B-EDUCATION": 10,
|
64 |
-
"B-EMAIL":
|
65 |
-
"B-EVENT":
|
66 |
"B-GENDER": 14,
|
67 |
-
"B-LAW":
|
68 |
-
"B-LOCATION":
|
69 |
-
"B-MISCELLANEOUS": 38,
|
70 |
-
"B-MONEY": 36,
|
71 |
"B-ORGANIZATION": 0,
|
72 |
-
"B-PERSON":
|
73 |
"B-PHONE": 18,
|
74 |
"B-PRODUCT": 8,
|
75 |
-
"B-QUANTITY":
|
76 |
"B-ROLE": 12,
|
77 |
"B-TRANSPORTATION": 13,
|
78 |
"B-URL": 11,
|
79 |
"I-ACTION": 17,
|
80 |
-
"I-AGE":
|
81 |
-
"I-DATETIME":
|
82 |
"I-DISEASE": 6,
|
83 |
"I-EDUCATION": 2,
|
84 |
"I-EMAIL": 28,
|
85 |
"I-EVENT": 4,
|
86 |
"I-GENDER": 1,
|
87 |
"I-LAW": 5,
|
88 |
-
"I-LOCATION":
|
89 |
-
"I-
|
90 |
-
"I-MONEY": 37,
|
91 |
-
"I-ORGANIZATION": 25,
|
92 |
"I-PERSON": 3,
|
93 |
-
"I-PHONE":
|
94 |
-
"I-PRODUCT":
|
95 |
-
"I-QUANTITY": 34,
|
96 |
"I-ROLE": 9,
|
97 |
"I-TRANSPORTATION": 27,
|
98 |
"I-URL": 16,
|
99 |
-
"O":
|
100 |
},
|
101 |
"layer_norm_eps": 1e-12,
|
102 |
"max_position_embeddings": 512,
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "NlpHUST/ner-vietnamese-electra-base",
|
3 |
"architectures": [
|
4 |
"ElectraForTokenClassification"
|
5 |
],
|
|
|
30 |
"16": "I-URL",
|
31 |
"17": "I-ACTION",
|
32 |
"18": "B-PHONE",
|
33 |
+
"19": "I-AGE",
|
34 |
+
"20": "O",
|
35 |
+
"21": "I-PRODUCT",
|
36 |
+
"22": "B-EVENT",
|
37 |
+
"23": "B-ACTION",
|
38 |
+
"24": "I-ORGANIZATION",
|
39 |
+
"25": "B-PERSON",
|
40 |
+
"26": "B-AGE",
|
41 |
"27": "I-TRANSPORTATION",
|
42 |
"28": "I-EMAIL",
|
43 |
+
"29": "B-EMAIL",
|
44 |
+
"30": "B-QUANTITY-NUMBER",
|
45 |
+
"31": "I-LOCATION",
|
46 |
+
"32": "I-DATETIME",
|
47 |
+
"33": "B-LOCATION",
|
48 |
+
"34": "I-QUANTITY-NUMBER",
|
49 |
+
"35": "I-PHONE",
|
50 |
+
"36": "B-LAW"
|
|
|
|
|
|
|
|
|
51 |
},
|
52 |
"initializer_range": 0.02,
|
53 |
"intermediate_size": 3072,
|
54 |
"label2id": {
|
55 |
+
"B-ACTION": 23,
|
56 |
+
"B-AGE": 26,
|
57 |
"B-DATETIME": 7,
|
58 |
"B-DISEASE": 15,
|
59 |
"B-EDUCATION": 10,
|
60 |
+
"B-EMAIL": 29,
|
61 |
+
"B-EVENT": 22,
|
62 |
"B-GENDER": 14,
|
63 |
+
"B-LAW": 36,
|
64 |
+
"B-LOCATION": 33,
|
|
|
|
|
65 |
"B-ORGANIZATION": 0,
|
66 |
+
"B-PERSON": 25,
|
67 |
"B-PHONE": 18,
|
68 |
"B-PRODUCT": 8,
|
69 |
+
"B-QUANTITY-NUMBER": 30,
|
70 |
"B-ROLE": 12,
|
71 |
"B-TRANSPORTATION": 13,
|
72 |
"B-URL": 11,
|
73 |
"I-ACTION": 17,
|
74 |
+
"I-AGE": 19,
|
75 |
+
"I-DATETIME": 32,
|
76 |
"I-DISEASE": 6,
|
77 |
"I-EDUCATION": 2,
|
78 |
"I-EMAIL": 28,
|
79 |
"I-EVENT": 4,
|
80 |
"I-GENDER": 1,
|
81 |
"I-LAW": 5,
|
82 |
+
"I-LOCATION": 31,
|
83 |
+
"I-ORGANIZATION": 24,
|
|
|
|
|
84 |
"I-PERSON": 3,
|
85 |
+
"I-PHONE": 35,
|
86 |
+
"I-PRODUCT": 21,
|
87 |
+
"I-QUANTITY-NUMBER": 34,
|
88 |
"I-ROLE": 9,
|
89 |
"I-TRANSPORTATION": 27,
|
90 |
"I-URL": 16,
|
91 |
+
"O": 20
|
92 |
},
|
93 |
"layer_norm_eps": 1e-12,
|
94 |
"max_position_embeddings": 512,
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f39c5b408664785f3d03989591503f1ccf15eda8d6a5fd6fcdcd0543f2b4117e
|
3 |
+
size 532404756
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 5112
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:16e513e54cbb50baa893755d2671e1b0e27f83c73dbc9d544aa57d433cabaf9b
|
3 |
size 5112
|