Replace toxicity model with multi-label classifier

This commit is contained in:
Jordan Wages 2026-08-26 17:00:49 -05:00
commit 1568c787f8
24 changed files with 2581129 additions and 119717 deletions

View file

@ -1,15 +1,19 @@
{
"source": "wagesj45/toxic-comment-classifier",
"revision": "a7d2df2ead42f0bce00b330939574a02266772f5",
"architecture": "DistilBertForSequenceClassification",
"source": "wagesj45/multilabel-toxic-comment-classifier",
"revision": "d90fd72e603240957e36d6afc8829d6988decff6",
"architecture": "ModernBertForSequenceClassification",
"labels": {
"toxic": 1,
"nonToxic": 0,
"names": {
"not_toxic": 0,
"toxic": 1
"toxicity": 0,
"severe_toxicity": 1,
"obscene": 2,
"threat": 3,
"insult": 4,
"identity_attack": 5,
"sexual_explicit": 6
}
},
"scoring": "multi-label-sigmoid",
"maxLength": 512,
"quantization": "int8-dynamic",
"runtime": "onnxruntime-web-wasm",