calpt commited on
Commit
da8d31e
1 Parent(s): 07518b2

Fix label2id mapping.

Browse files
Files changed (1) hide show
  1. head_config.json +23 -23
head_config.json CHANGED
@@ -3,29 +3,29 @@
3
  "activation_function": "tanh",
4
  "head_type": "tagging",
5
  "label2id": {
6
- "LABEL_0": 0,
7
- "LABEL_1": 1,
8
- "LABEL_10": 10,
9
- "LABEL_11": 11,
10
- "LABEL_12": 12,
11
- "LABEL_13": 13,
12
- "LABEL_14": 14,
13
- "LABEL_15": 15,
14
- "LABEL_16": 16,
15
- "LABEL_17": 17,
16
- "LABEL_18": 18,
17
- "LABEL_19": 19,
18
- "LABEL_2": 2,
19
- "LABEL_20": 20,
20
- "LABEL_21": 21,
21
- "LABEL_22": 22,
22
- "LABEL_3": 3,
23
- "LABEL_4": 4,
24
- "LABEL_5": 5,
25
- "LABEL_6": 6,
26
- "LABEL_7": 7,
27
- "LABEL_8": 8,
28
- "LABEL_9": 9
29
  },
30
  "layers": 1,
31
  "num_labels": 23
3
  "activation_function": "tanh",
4
  "head_type": "tagging",
5
  "label2id": {
6
+ "O": 0,
7
+ "B-ADJP": 1,
8
+ "I-ADJP": 2,
9
+ "B-ADVP": 3,
10
+ "I-ADVP": 4,
11
+ "B-CONJP": 5,
12
+ "I-CONJP": 6,
13
+ "B-INTJ": 7,
14
+ "I-INTJ": 8,
15
+ "B-LST": 9,
16
+ "I-LST": 10,
17
+ "B-NP": 11,
18
+ "I-NP": 12,
19
+ "B-PP": 13,
20
+ "I-PP": 14,
21
+ "B-PRT": 15,
22
+ "I-PRT": 16,
23
+ "B-SBAR": 17,
24
+ "I-SBAR": 18,
25
+ "B-UCP": 19,
26
+ "I-UCP": 20,
27
+ "B-VP": 21,
28
+ "I-VP": 22
29
  },
30
  "layers": 1,
31
  "num_labels": 23