Danil commited on
Commit
7254e36
1 Parent(s): 93c8f24

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -16,6 +16,10 @@
16
  "max_position_embeddings": 514,
17
  "model_type": "roberta",
18
  "num_attention_heads": 12,
 
 
 
 
19
  "num_hidden_layers": 12,
20
  "pad_token_id": 1,
21
  "position_embedding_type": "absolute",
 
16
  "max_position_embeddings": 514,
17
  "model_type": "roberta",
18
  "num_attention_heads": 12,
19
+ "id2label": {
20
+ "0": "NOT_SPAM",
21
+ "1": "SPAM"
22
+ },
23
  "num_hidden_layers": 12,
24
  "pad_token_id": 1,
25
  "position_embedding_type": "absolute",