calpt commited on
Commit
db0cd07
1 Parent(s): 5d7951a

Fix label2id mapping.

Browse files
Files changed (1) hide show
  1. head_config.json +6 -5
head_config.json CHANGED
@@ -3,11 +3,12 @@
3
  "activation_function": "tanh",
4
  "head_type": "multiple_choice",
5
  "label2id": {
6
- "LABEL_0": 0,
7
- "LABEL_1": 1,
8
- "LABEL_2": 2,
9
- "LABEL_3": 3,
10
- "LABEL_4": 4
 
11
  },
12
  "layers": 2,
13
  "num_choices": 5,
 
3
  "activation_function": "tanh",
4
  "head_type": "multiple_choice",
5
  "label2id": {
6
+ "": 0,
7
+ "A": 1,
8
+ "B": 2,
9
+ "C": 3,
10
+ "D": 4,
11
+ "E": 5
12
  },
13
  "layers": 2,
14
  "num_choices": 5,