update thresholds
Browse files- config.json +21 -21
config.json
CHANGED
@@ -20,9 +20,9 @@
|
|
20 |
"hateful": {
|
21 |
"calibrated": false,
|
22 |
"threshold": {
|
23 |
-
"high_recall": 0.
|
24 |
-
"balanced": 0.
|
25 |
-
"high_precision":
|
26 |
},
|
27 |
"model_type": "ridge_classifier",
|
28 |
"model_fp": "models/lionguard-hateful.onnx"
|
@@ -30,9 +30,9 @@
|
|
30 |
"harassment": {
|
31 |
"calibrated": false,
|
32 |
"threshold": {
|
33 |
-
"high_recall":
|
34 |
-
"balanced":
|
35 |
-
"high_precision":
|
36 |
},
|
37 |
"model_type": "ridge_classifier",
|
38 |
"model_fp": "models/lionguard-harassment.onnx"
|
@@ -40,9 +40,9 @@
|
|
40 |
"public_harm": {
|
41 |
"calibrated": false,
|
42 |
"threshold": {
|
43 |
-
"high_recall": 0.
|
44 |
-
"balanced": 0.
|
45 |
-
"high_precision": 0.
|
46 |
},
|
47 |
"model_type": "ridge_classifier",
|
48 |
"model_fp": "models/lionguard-public_harm.onnx"
|
@@ -50,9 +50,9 @@
|
|
50 |
"self_harm": {
|
51 |
"calibrated": false,
|
52 |
"threshold": {
|
53 |
-
"high_recall": 0.
|
54 |
-
"balanced": 0.
|
55 |
-
"high_precision": 0.
|
56 |
},
|
57 |
"model_type": "ridge_classifier",
|
58 |
"model_fp": "models/lionguard-self_harm.onnx"
|
@@ -60,9 +60,9 @@
|
|
60 |
"sexual": {
|
61 |
"calibrated": false,
|
62 |
"threshold": {
|
63 |
-
"high_recall": 0.
|
64 |
-
"balanced": 0.
|
65 |
-
"high_precision": 0.
|
66 |
},
|
67 |
"model_type": "ridge_classifier",
|
68 |
"model_fp": "models/lionguard-sexual.onnx"
|
@@ -70,9 +70,9 @@
|
|
70 |
"toxic": {
|
71 |
"calibrated": false,
|
72 |
"threshold": {
|
73 |
-
"high_recall":
|
74 |
-
"balanced": 0.
|
75 |
-
"high_precision": 0.
|
76 |
},
|
77 |
"model_type": "ridge_classifier",
|
78 |
"model_fp": "models/lionguard-toxic.onnx"
|
@@ -80,9 +80,9 @@
|
|
80 |
"violent": {
|
81 |
"calibrated": false,
|
82 |
"threshold": {
|
83 |
-
"high_recall": 0.
|
84 |
-
"balanced": 0.
|
85 |
-
"high_precision": 0.
|
86 |
},
|
87 |
"model_type": "ridge_classifier",
|
88 |
"model_fp": "models/lionguard-violent.onnx"
|
|
|
20 |
"hateful": {
|
21 |
"calibrated": false,
|
22 |
"threshold": {
|
23 |
+
"high_recall": -0.341,
|
24 |
+
"balanced": -0.186,
|
25 |
+
"high_precision": -0.008
|
26 |
},
|
27 |
"model_type": "ridge_classifier",
|
28 |
"model_fp": "models/lionguard-hateful.onnx"
|
|
|
30 |
"harassment": {
|
31 |
"calibrated": false,
|
32 |
"threshold": {
|
33 |
+
"high_recall": -0.571,
|
34 |
+
"balanced": -0.471,
|
35 |
+
"high_precision": -0.471
|
36 |
},
|
37 |
"model_type": "ridge_classifier",
|
38 |
"model_fp": "models/lionguard-harassment.onnx"
|
|
|
40 |
"public_harm": {
|
41 |
"calibrated": false,
|
42 |
"threshold": {
|
43 |
+
"high_recall": -0.713,
|
44 |
+
"balanced": -0.632,
|
45 |
+
"high_precision": -0.576
|
46 |
},
|
47 |
"model_type": "ridge_classifier",
|
48 |
"model_fp": "models/lionguard-public_harm.onnx"
|
|
|
50 |
"self_harm": {
|
51 |
"calibrated": false,
|
52 |
"threshold": {
|
53 |
+
"high_recall": -0.663,
|
54 |
+
"balanced": -0.617,
|
55 |
+
"high_precision": -0.608
|
56 |
},
|
57 |
"model_type": "ridge_classifier",
|
58 |
"model_fp": "models/lionguard-self_harm.onnx"
|
|
|
60 |
"sexual": {
|
61 |
"calibrated": false,
|
62 |
"threshold": {
|
63 |
+
"high_recall": -0.234,
|
64 |
+
"balanced": 0.044,
|
65 |
+
"high_precision": 0.277
|
66 |
},
|
67 |
"model_type": "ridge_classifier",
|
68 |
"model_fp": "models/lionguard-sexual.onnx"
|
|
|
70 |
"toxic": {
|
71 |
"calibrated": false,
|
72 |
"threshold": {
|
73 |
+
"high_recall": 0.098,
|
74 |
+
"balanced": 0.315,
|
75 |
+
"high_precision": 0.523
|
76 |
},
|
77 |
"model_type": "ridge_classifier",
|
78 |
"model_fp": "models/lionguard-toxic.onnx"
|
|
|
80 |
"violent": {
|
81 |
"calibrated": false,
|
82 |
"threshold": {
|
83 |
+
"high_recall": -0.505,
|
84 |
+
"balanced": -0.343,
|
85 |
+
"high_precision": -0.343
|
86 |
},
|
87 |
"model_type": "ridge_classifier",
|
88 |
"model_fp": "models/lionguard-violent.onnx"
|