calpt commited on
Commit
e9517bc
1 Parent(s): 9a35df3

Fix label2id mapping.

Browse files
Files changed (1) hide show
  1. head_config.json +7 -7
head_config.json CHANGED
@@ -3,15 +3,15 @@
3
  "activation_function": "tanh",
4
  "head_type": "tagging",
5
  "label2id": {
6
- "B-LOC": 5,
7
- "B-MISC": 7,
8
- "B-ORG": 3,
9
  "B-PER": 1,
10
- "I-LOC": 6,
11
- "I-MISC": 8,
12
- "I-ORG": 4,
13
  "I-PER": 2,
14
- "O": 0
 
 
 
 
 
15
  },
16
  "layers": 1,
17
  "num_labels": 9
3
  "activation_function": "tanh",
4
  "head_type": "tagging",
5
  "label2id": {
6
+ "O": 0,
 
 
7
  "B-PER": 1,
 
 
 
8
  "I-PER": 2,
9
+ "B-ORG": 3,
10
+ "I-ORG": 4,
11
+ "B-LOC": 5,
12
+ "I-LOC": 6,
13
+ "B-MISC": 7,
14
+ "I-MISC": 8
15
  },
16
  "layers": 1,
17
  "num_labels": 9