CIS519_NEWS_Classify / config.json
Eric22333's picture
Update config.json
705d0d7 verified
raw
history blame contribute delete
282 Bytes
{
"architectures": ["NewsClassifier"],
"model_type": "custom",
"max_length": 41,
"vocab_size": 74934,
"embedding_dim": 128,
"hidden_size": 64,
"num_labels": 2,
"id2label": {
"0": "nbc",
"1": "foxnews"
},
"label2id": {
"nbc": 0,
"foxnews": 1
}
}