File size: 2,855 Bytes
97f9a3c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8e8fa11
97f9a3c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
{
    "description": "Binary classifier on harmful text in Singapore context",
    "embedding": {
        "tokenizer": "BAAI/bge-large-en-v1.5",
        "model": "BAAI/bge-large-en-v1.5",
        "max_length": 512,
        "batch_size": 32
    },
    "classifier": {
        "binary": {
            "calibrated": true,
            "threshold": {
                "high_recall": 0.2,
                "balanced": 0.5,
                "high_precision": 0.8
            },
            "model_type": "ridge_classifier",
            "model_fp": "models/lionguard-binary.onnx"
        },
        "hateful": {
            "calibrated": false,
            "threshold": {
                "high_recall": 0.516,
                "balanced": 0.827,
                "high_precision": 1.254
            },
            "model_type": "ridge_classifier",
            "model_fp": "models/lionguard-hateful.onnx"
        },
        "harassment": {
            "calibrated": false,
            "threshold": {
                "high_recall": 1.326,
                "balanced": 1.326,
                "high_precision": 1.955
            },
            "model_type": "ridge_classifier",
            "model_fp": "models/lionguard-harassment.onnx"
        },
        "public_harm": {
            "calibrated": false,
            "threshold": {
                "high_recall": 0.953,
                "balanced": 0.953,
                "high_precision": 0.953
            },
            "model_type": "ridge_classifier",
            "model_fp": "models/lionguard-public_harm.onnx"
        },
        "self_harm": {
            "calibrated": false,
            "threshold": {
                "high_recall": 0.915,
                "balanced": 0.915,
                "high_precision": 0.915
            },
            "model_type": "ridge_classifier",
            "model_fp": "models/lionguard-self_harm.onnx"
        },
        "sexual": {
            "calibrated": false,
            "threshold": {
                "high_recall": 0.388,
                "balanced": 0.500,
                "high_precision": 0.702
            },
            "model_type": "ridge_classifier",
            "model_fp": "models/lionguard-sexual.onnx"
        },
        "toxic": {
            "calibrated": false,
            "threshold": {
                "high_recall": -0.089,
                "balanced": 0.136,
                "high_precision": 0.327
            },
            "model_type": "ridge_classifier",
            "model_fp": "models/lionguard-toxic.onnx"
        },
        "violent": {
            "calibrated": false,
            "threshold": {
                "high_recall": 0.317,
                "balanced": 0.981,
                "high_precision": 0.981
            },
            "model_type": "ridge_classifier",
            "model_fp": "models/lionguard-violent.onnx"
        }
    }
  }