calpt commited on
Commit
9206193
1 Parent(s): 5f271fa

Fix label2id mapping.

Browse files
Files changed (1) hide show
  1. head_config.json +18 -18
head_config.json CHANGED
@@ -3,24 +3,24 @@
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_2": 2,
17
- "LABEL_3": 3,
18
- "LABEL_4": 4,
19
- "LABEL_5": 5,
20
- "LABEL_6": 6,
21
- "LABEL_7": 7,
22
- "LABEL_8": 8,
23
- "LABEL_9": 9
24
  },
25
  "layers": 1,
26
  "num_labels": 18
3
  "activation_function": "tanh",
4
  "head_type": "tagging",
5
  "label2id": {
6
+ "NOUN": 0,
7
+ "PUNCT": 1,
8
+ "ADP": 2,
9
+ "NUM": 3,
10
+ "SYM": 4,
11
+ "SCONJ": 5,
12
+ "ADJ": 6,
13
+ "PART": 7,
14
+ "DET": 8,
15
+ "CCONJ": 9,
16
+ "PROPN": 10,
17
+ "PRON": 11,
18
+ "X": 12,
19
+ "_": 13,
20
+ "ADV": 14,
21
+ "INTJ": 15,
22
+ "VERB": 16,
23
+ "AUX": 17
24
  },
25
  "layers": 1,
26
  "num_labels": 18