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"
         | 
