Text Classification
Transformers
ONNX
Safetensors
bert
toxicity
bert-tiny
gravitee-io
ai-gateway
remisultan commited on
Commit
627353e
·
1 Parent(s): 86ef183

feat: add id2label and label2id config attributes

Browse files
Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -17,6 +17,14 @@
17
  "pad_token_id": 0,
18
  "position_embedding_type": "absolute",
19
  "problem_type": "single_label_classification",
 
 
 
 
 
 
 
 
20
  "torch_dtype": "float32",
21
  "transformers_version": "4.53.2",
22
  "type_vocab_size": 2,
 
17
  "pad_token_id": 0,
18
  "position_embedding_type": "absolute",
19
  "problem_type": "single_label_classification",
20
+ "id2label": {
21
+ "0": "not-toxic",
22
+ "1": "toxic"
23
+ },
24
+ "label2id": {
25
+ "not-toxic": 0,
26
+ "toxic": 1
27
+ },
28
  "torch_dtype": "float32",
29
  "transformers_version": "4.53.2",
30
  "type_vocab_size": 2,