JungleLee commited on
Commit
6b378ab
1 Parent(s): b1ff513

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -2
config.json CHANGED
@@ -21,8 +21,15 @@
21
  "problem_type": "single_label_classification",
22
  "return_dict": false,
23
  "torch_dtype": "float32",
24
- "transformers_version": "4.13.0",
25
  "type_vocab_size": 2,
26
  "use_cache": true,
27
- "vocab_size": 30522
 
 
 
 
 
 
 
 
28
  }
 
21
  "problem_type": "single_label_classification",
22
  "return_dict": false,
23
  "torch_dtype": "float32",
 
24
  "type_vocab_size": 2,
25
  "use_cache": true,
26
+ "vocab_size": 30522,
27
+ "id2label": {
28
+ "0": "non-toxic",
29
+ "1": "toxic"
30
+ },
31
+ "label2id": {
32
+ "non-toxic": 0,
33
+ "toxic": 1
34
+ }
35
  }