avialfont commited on
Commit
ed687be
1 Parent(s): 36f9084

add config

Browse files
Files changed (1) hide show
  1. config.json +12 -12
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "bert-base-cased",
3
  "architectures": [
4
  "BertForTokenClassification"
5
  ],
@@ -11,8 +11,8 @@
11
  "hidden_size": 768,
12
  "id2label": {
13
  "0": "O",
14
- "1": "B-PER",
15
- "2": "I-PER",
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": "5",
27
- "B-MISC": "7",
28
- "B-ORG": "3",
29
- "B-PER": "1",
30
- "I-LOC": "6",
31
- "I-MISC": "8",
32
- "I-ORG": "4",
33
- "I-PER": "2",
34
- "O": "0"
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,