Upload RobertaForSequenceClassification

README.md

@@ -1,13 +1,13 @@
 ---
 language:
 - en
-widget:
-- text: Call me today to earn some money mofos!
+license: mit
 datasets:
 - cardiffnlp/x_sensitive
-license: mit
 metrics:
 - f1
+widget:
+- text: Call me today to earn some money mofos!
 pipeline_tag: text-classification
 ---
 

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/home/dimosthenisa/Desktop/sensitive_content/code/experiments/mlm_models/troberta-large/multilabel/best_model",
+  "_name_or_path": "/home/dimosthenisa/Desktop/sensitive_content/code/experiments/mlm_models/troberta-large_seven-labels/multilabel/best_model",
   "architectures": [
     "RobertaForSequenceClassification"
   ],
@@ -16,13 +16,15 @@
     "2": "sex",
     "3": "drugs",
     "4": "selfharm",
-    "5": "spam"
+    "5": "spam",
+    "6": "not-sensitive"
   },
   "initializer_range": 0.02,
   "intermediate_size": 4096,
   "label2id": {
     "conflictual": 0,
     "drugs": 3,
+    "not-sensitive": 6,
     "profanity": 1,
     "selfharm": 4,
     "sex": 2,
@@ -37,7 +39,7 @@
   "position_embedding_type": "absolute",
   "problem_type": "multi_label_classification",
   "torch_dtype": "float32",
-  "transformers_version": "4.41.2",
+  "transformers_version": "4.39.3",
   "type_vocab_size": 1,
   "use_cache": true,
   "vocab_size": 50265

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6f5e6d0e4f8e4af5a9a9a75f7f16cd15644ea235e5c7240c97fa650489fea0ec
-size 1421511816
+oid sha256:e20705b5f23678bf6b0f51de8711bdb16e56e25901575b689a6f52f1de861d11
+size 1421515916