SupratimRK's picture
Sync model from GitHub repository
c801521 verified
Raw History Blame Contribute Delete
482 Bytes
{
"model_type": "CompactNameClassifier",
"architectures": [
"CompactNameClassifier"
],
"vocab_size": 4000,
"emb_dim": 64,
"hidden_dim": 128,
"num_layers": 2,
"num_heads": 4,
"dim_feedforward": 256,
"max_length": 20,
"dropout": 0.1,
"id2label": {
"0": "NON_NAME",
"1": "REAL_NAME"
},
"label2id": {
"NON_NAME": 0,
"REAL_NAME": 1
},
"metrics": {
"best_val_accuracy": 0.942580927678782,
"best_val_f1": 0.9606740276938268
}
}