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