ikuyamada commited on
Commit
eb1d041
1 Parent(s): 5f7f060

add id2label and label2id to config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -11
config.json CHANGED
@@ -3,7 +3,6 @@
3
  "LukeForEntitySpanClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
- "bert_model_name": "roberta-large",
7
  "bos_token_id": 0,
8
  "entity_emb_size": 256,
9
  "entity_vocab_size": 500000,
@@ -13,20 +12,20 @@
13
  "hidden_dropout_prob": 0.1,
14
  "hidden_size": 1024,
15
  "id2label": {
16
- "0": "LABEL_0",
17
- "1": "LABEL_1",
18
- "2": "LABEL_2",
19
- "3": "LABEL_3",
20
- "4": "LABEL_4"
21
  },
22
  "initializer_range": 0.02,
23
  "intermediate_size": 4096,
24
  "label2id": {
25
- "LABEL_0": 0,
26
- "LABEL_1": 1,
27
- "LABEL_2": 2,
28
- "LABEL_3": 3,
29
- "LABEL_4": 4
30
  },
31
  "layer_norm_eps": 1e-05,
32
  "max_position_embeddings": 514,
 
3
  "LukeForEntitySpanClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
 
6
  "bos_token_id": 0,
7
  "entity_emb_size": 256,
8
  "entity_vocab_size": 500000,
 
12
  "hidden_dropout_prob": 0.1,
13
  "hidden_size": 1024,
14
  "id2label": {
15
+ "0": "NIL",
16
+ "1": "MISC",
17
+ "2": "PER",
18
+ "3": "ORG",
19
+ "4": "LOC"
20
  },
21
  "initializer_range": 0.02,
22
  "intermediate_size": 4096,
23
  "label2id": {
24
+ "LOC": 4,
25
+ "MISC": 1,
26
+ "NIL": 0,
27
+ "ORG": 3,
28
+ "PER": 2
29
  },
30
  "layer_norm_eps": 1e-05,
31
  "max_position_embeddings": 514,