calpt commited on
Commit
4e3ef68
1 Parent(s): b7a6e66

Fix label2id mapping.

Browse files
Files changed (1) hide show
  1. head_config.json +7 -7
head_config.json CHANGED
@@ -3,19 +3,19 @@
3
  "activation_function": "tanh",
4
  "head_type": "tagging",
5
  "label2id": {
 
6
  "B-corporation": 1,
7
- "B-creative-work": 3,
8
- "B-group": 5,
9
- "B-location": 7,
10
- "B-person": 9,
11
- "B-product": 11,
12
  "I-corporation": 2,
 
13
  "I-creative-work": 4,
 
14
  "I-group": 6,
 
15
  "I-location": 8,
 
16
  "I-person": 10,
17
- "I-product": 12,
18
- "O": 0
19
  },
20
  "layers": 1,
21
  "num_labels": 13
 
3
  "activation_function": "tanh",
4
  "head_type": "tagging",
5
  "label2id": {
6
+ "O": 0,
7
  "B-corporation": 1,
 
 
 
 
 
8
  "I-corporation": 2,
9
+ "B-creative-work": 3,
10
  "I-creative-work": 4,
11
+ "B-group": 5,
12
  "I-group": 6,
13
+ "B-location": 7,
14
  "I-location": 8,
15
+ "B-person": 9,
16
  "I-person": 10,
17
+ "B-product": 11,
18
+ "I-product": 12
19
  },
20
  "layers": 1,
21
  "num_labels": 13