roberta-base-pf-ud_pos / head_config.json
calpt's picture
Fix label2id mapping.
9206193
{
"config": {
"activation_function": "tanh",
"head_type": "tagging",
"label2id": {
"NOUN": 0,
"PUNCT": 1,
"ADP": 2,
"NUM": 3,
"SYM": 4,
"SCONJ": 5,
"ADJ": 6,
"PART": 7,
"DET": 8,
"CCONJ": 9,
"PROPN": 10,
"PRON": 11,
"X": 12,
"_": 13,
"ADV": 14,
"INTJ": 15,
"VERB": 16,
"AUX": 17
},
"layers": 1,
"num_labels": 18
},
"hidden_size": 768,
"model_class": "RobertaModelWithHeads",
"model_name": "roberta-base",
"model_type": "roberta",
"name": "universal_dependencies_en_ewt"
}