TypeBERT / config.json
Guihss's picture
Upload TypeBERTForSequenceClassification
a9c0c7a
raw
history blame
621 Bytes
{
"architectures": [
"TypeBERTForSequenceClassification"
],
"auto_map": {
"AutoConfig": "type_bert_model.TypeBERTConfig",
"AutoModelForSequenceClassification": "type_bert_model.TypeBERTForSequenceClassification"
},
"id2label": {
"0": "agent",
"1": "event",
"2": "place",
"3": "item",
"4": "virtual",
"5": "concept"
},
"label2id": {
"agent": 0,
"concept": 5,
"event": 1,
"item": 3,
"place": 2,
"virtual": 4
},
"model_type": "type_bert",
"tokenizer_class": "bert-base-uncased",
"torch_dtype": "float32",
"transformers_version": "4.22.1"
}