roberta-base-pf-wnut_17 / head_config.json
calpt's picture
Fix label2id mapping.
b6f6ebc
{
"config": {
"activation_function": "tanh",
"head_type": "tagging",
"label2id": {
"O": 0,
"B-corporation": 1,
"I-corporation": 2,
"B-creative-work": 3,
"I-creative-work": 4,
"B-group": 5,
"I-group": 6,
"B-location": 7,
"I-location": 8,
"B-person": 9,
"I-person": 10,
"B-product": 11,
"I-product": 12
},
"layers": 1,
"num_labels": 13
},
"hidden_size": 768,
"model_class": "RobertaModelWithHeads",
"model_name": "roberta-base",
"model_type": "roberta",
"name": "wnut_17"
}