add config
Browse files- config.json +12 -12
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"BertForTokenClassification"
|
5 |
],
|
@@ -11,8 +11,8 @@
|
|
11 |
"hidden_size": 768,
|
12 |
"id2label": {
|
13 |
"0": "O",
|
14 |
-
"1": "B-
|
15 |
-
"2": "I-
|
16 |
"3": "B-ORG",
|
17 |
"4": "I-ORG",
|
18 |
"5": "B-LOC",
|
@@ -23,15 +23,15 @@
|
|
23 |
"initializer_range": 0.02,
|
24 |
"intermediate_size": 3072,
|
25 |
"label2id": {
|
26 |
-
"B-LOC":
|
27 |
-
"B-MISC":
|
28 |
-
"B-ORG":
|
29 |
-
"B-
|
30 |
-
"I-LOC":
|
31 |
-
"I-MISC":
|
32 |
-
"I-ORG":
|
33 |
-
"I-
|
34 |
-
"O":
|
35 |
},
|
36 |
"layer_norm_eps": 1e-12,
|
37 |
"max_position_embeddings": 512,
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "./saved_models/token_classification",
|
3 |
"architectures": [
|
4 |
"BertForTokenClassification"
|
5 |
],
|
|
|
11 |
"hidden_size": 768,
|
12 |
"id2label": {
|
13 |
"0": "O",
|
14 |
+
"1": "B-SM1",
|
15 |
+
"2": "I-SM1",
|
16 |
"3": "B-ORG",
|
17 |
"4": "I-ORG",
|
18 |
"5": "B-LOC",
|
|
|
23 |
"initializer_range": 0.02,
|
24 |
"intermediate_size": 3072,
|
25 |
"label2id": {
|
26 |
+
"B-LOC": 5,
|
27 |
+
"B-MISC": 7,
|
28 |
+
"B-ORG": 3,
|
29 |
+
"B-SM1": 1,
|
30 |
+
"I-LOC": 6,
|
31 |
+
"I-MISC": 8,
|
32 |
+
"I-ORG": 4,
|
33 |
+
"I-SM1": 2,
|
34 |
+
"O": 0
|
35 |
},
|
36 |
"layer_norm_eps": 1e-12,
|
37 |
"max_position_embeddings": 512,
|