laurahanu commited on
Commit
3e3a905
·
1 Parent(s): 85c360d

update label names

Browse files
Files changed (1) hide show
  1. config.json +16 -16
config.json CHANGED
@@ -30,22 +30,22 @@
30
  "initializer_range": 0.02,
31
  "intermediate_size": 3072,
32
  "label2id": {
33
- "LABEL_0": 0,
34
- "LABEL_1": 1,
35
- "LABEL_10": 10,
36
- "LABEL_11": 11,
37
- "LABEL_12": 12,
38
- "LABEL_13": 13,
39
- "LABEL_14": 14,
40
- "LABEL_15": 15,
41
- "LABEL_2": 2,
42
- "LABEL_3": 3,
43
- "LABEL_4": 4,
44
- "LABEL_5": 5,
45
- "LABEL_6": 6,
46
- "LABEL_7": 7,
47
- "LABEL_8": 8,
48
- "LABEL_9": 9
49
  },
50
  "layer_norm_eps": 1e-05,
51
  "max_position_embeddings": 514,
 
30
  "initializer_range": 0.02,
31
  "intermediate_size": 3072,
32
  "label2id": {
33
+ "toxicity": 0,
34
+ "severe_toxicity": 1,
35
+ "christian": 10,
36
+ "jewish": 11,
37
+ "muslim": 12,
38
+ "black": 13,
39
+ "white": 14,
40
+ "psychiatric_or_mental_illness": 15,
41
+ "obscene": 2,
42
+ "identity_attack": 3,
43
+ "insult": 4,
44
+ "threat": 5,
45
+ "sexual_explicit": 6,
46
+ "male": 7,
47
+ "female": 8,
48
+ "homosexual_gay_or_lesbian": 9
49
  },
50
  "layer_norm_eps": 1e-05,
51
  "max_position_embeddings": 514,