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-base_seven-labels/multilabel/best_model",
   "architectures": [
     "RobertaForSequenceClassification"
   ],
@@ -9,20 +9,22 @@
   "eos_token_id": 2,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
-  "hidden_size": 1024,
+  "hidden_size": 768,
   "id2label": {
     "0": "conflictual",
     "1": "profanity",
     "2": "sex",
     "3": "drugs",
     "4": "selfharm",
-    "5": "spam"
+    "5": "spam",
+    "6": "not-sensitive"
   },
   "initializer_range": 0.02,
-  "intermediate_size": 4096,
+  "intermediate_size": 3072,
   "label2id": {
     "conflictual": 0,
     "drugs": 3,
+    "not-sensitive": 6,
     "profanity": 1,
     "selfharm": 4,
     "sex": 2,
@@ -31,13 +33,13 @@
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,
   "model_type": "roberta",
-  "num_attention_heads": 16,
-  "num_hidden_layers": 24,
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
   "pad_token_id": 1,
   "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:6c19bbdab3c5346a48db36d125c579a30ab8b579668064b2d335d7f376c8ee4b
+size 498628204