Replace toxicity model with multi-label classifier
This commit is contained in:
parent
803df7feee
commit
1568c787f8
24 changed files with 2581129 additions and 119717 deletions
|
|
@ -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",
|
||||
|
|
|
|||
Loading…
Reference in a new issue