update label names
Browse files- 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": "
|
14 |
-
"1": "
|
15 |
-
"2": "
|
16 |
-
"3": "
|
17 |
-
"4": "
|
18 |
-
"5": "
|
19 |
-
"6": "
|
20 |
-
"7": "
|
21 |
-
"8": "
|
22 |
-
"9": "
|
23 |
-
"10": "
|
24 |
-
"11": "
|
25 |
-
"12": "
|
26 |
-
"13": "
|
27 |
-
"14": "
|
28 |
-
"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,
|