citizenlab
/

distilbert-base-multilingual-cased-toxicity

Text Classification

Inference Endpoints

Model card Files Files and versions Community

arianpasquali commited on Jan 25, 2022

Commit

c070717

•

1 Parent(s): af6bf4c

add model

Files changed (2) hide show

config.json +2 -10
pytorch_model.bin +2 -2

config.json CHANGED Viewed

@@ -10,19 +10,11 @@
   "hidden_dim": 3072,
   "id2label": {
     "0": "toxic",
-    "1": "severe_toxic",
-    "2": "obscene",
-    "3": "threat",
-    "4": "insult",
-    "5": "identity_hate"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "identity_hate": 5,
-    "insult": 4,
-    "obscene": 2,
-    "severe_toxic": 1,
-    "threat": 3,
     "toxic": 0
   },
   "max_position_embeddings": 512,

   "hidden_dim": 3072,
   "id2label": {
     "0": "toxic",
+    "1": "not_toxic"
   },
   "initializer_range": 0.02,
   "label2id": {
+    "not_toxic": 1,
     "toxic": 0
   },
   "max_position_embeddings": 512,

pytorch_model.bin CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cd37fbde74f1dabda5121c478875539e6baa8c1ec440caa6fb7bc478de4e6dc4
-size 541357169

 version https://git-lfs.github.com/spec/v1
+oid sha256:97747b7be915228c122fa9745fda847e9af3a1dcd9c12ab97892a873db3f47bb
+size 541344881