Shoriful025's picture
Create config.json
d96106b verified
```json
{
"architectures": [
"RobertaForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"bos_token_id": 0,
"eos_token_id": 2,
"hidden_act": "gelu",
"hidden_size": 768,
"model_type": "roberta",
"num_attention_heads": 12,
"num_hidden_layers": 12,
"type_vocab_size": 1,
"vocab_size": 50265,
"id2label": {
"0": "BENIGN",
"1": "MALICIOUS"
},
"label2id": {
"BENIGN": 0,
"MALICIOUS": 1
}
}