shaunkhoo commited on
Commit
517f106
1 Parent(s): d964857

update thresholds

Browse files
Files changed (1) hide show
  1. config.json +21 -21
config.json CHANGED
@@ -20,9 +20,9 @@
20
  "hateful": {
21
  "calibrated": false,
22
  "threshold": {
23
- "high_recall": 0.516,
24
- "balanced": 0.827,
25
- "high_precision": 1.254
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": 1.326,
34
- "balanced": 1.326,
35
- "high_precision": 1.955
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.953,
44
- "balanced": 0.953,
45
- "high_precision": 0.953
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.915,
54
- "balanced": 0.915,
55
- "high_precision": 0.915
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.388,
64
- "balanced": 0.500,
65
- "high_precision": 0.702
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": -0.089,
74
- "balanced": 0.136,
75
- "high_precision": 0.327
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.317,
84
- "balanced": 0.981,
85
- "high_precision": 0.981
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"