laurahanu commited on
Commit
140db5f
1 Parent(s): 3e3a905

update label names

Browse files
Files changed (1) hide show
  1. config.json +16 -16
config.json CHANGED
@@ -10,22 +10,22 @@
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
- "0": "LABEL_0",
14
- "1": "LABEL_1",
15
- "2": "LABEL_2",
16
- "3": "LABEL_3",
17
- "4": "LABEL_4",
18
- "5": "LABEL_5",
19
- "6": "LABEL_6",
20
- "7": "LABEL_7",
21
- "8": "LABEL_8",
22
- "9": "LABEL_9",
23
- "10": "LABEL_10",
24
- "11": "LABEL_11",
25
- "12": "LABEL_12",
26
- "13": "LABEL_13",
27
- "14": "LABEL_14",
28
- "15": "LABEL_15"
29
  },
30
  "initializer_range": 0.02,
31
  "intermediate_size": 3072,
 
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
+ "0": "toxicity",
14
+ "1": "severe_toxicity",
15
+ "2": "obscene",
16
+ "3": "identity_attack",
17
+ "4": "insult",
18
+ "5": "threat",
19
+ "6": "sexual_explicit",
20
+ "7": "male",
21
+ "8": "female",
22
+ "9": "homosexual_gay_or_lesbian",
23
+ "10": "christian",
24
+ "11": "jewish",
25
+ "12": "muslim",
26
+ "13": "black",
27
+ "14": "white",
28
+ "15": "psychiatric_or_mental_illness"
29
  },
30
  "initializer_range": 0.02,
31
  "intermediate_size": 3072,