ZiruiXiong commited on
Commit
628fcde
·
1 Parent(s): 151ce4a

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -4
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "../../bert-base-uncased",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
@@ -18,10 +18,16 @@
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 0,
20
  "position_embedding_type": "absolute",
21
- "return_dict": false,
22
  "torch_dtype": "float32",
23
- "transformers_version": "4.13.0",
24
  "type_vocab_size": 2,
25
  "use_cache": true,
26
- "vocab_size": 30522
 
 
 
 
 
 
 
 
27
  }
 
1
  {
2
+ "_name_or_path": "bert-base-uncased",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
 
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 0,
20
  "position_embedding_type": "absolute",
 
21
  "torch_dtype": "float32",
 
22
  "type_vocab_size": 2,
23
  "use_cache": true,
24
+ "vocab_size": 30522,
25
+ "id2label": {
26
+ "0": "non-toxic",
27
+ "1": "toxic"
28
+ },
29
+ "label2id": {
30
+ "non-toxic": 0,
31
+ "toxic": 1
32
+ }
33
  }