calpt commited on
Commit
7b69de3
1 Parent(s): 692de14

Fix label2id mapping.

Browse files
Files changed (1) hide show
  1. head_config.json +6 -6
head_config.json CHANGED
@@ -4,12 +4,12 @@
4
  "bias": true,
5
  "head_type": "classification",
6
  "label2id": {
7
- "LABEL_0": 0,
8
- "LABEL_1": 1,
9
- "LABEL_2": 2,
10
- "LABEL_3": 3,
11
- "LABEL_4": 4,
12
- "LABEL_5": 5
13
  },
14
  "layers": 2,
15
  "num_labels": 6,
 
4
  "bias": true,
5
  "head_type": "classification",
6
  "label2id": {
7
+ "DESC": 0,
8
+ "ENTY": 1,
9
+ "ABBR": 2,
10
+ "HUM": 3,
11
+ "NUM": 4,
12
+ "LOC": 5
13
  },
14
  "layers": 2,
15
  "num_labels": 6,