Upload BertForSequenceClassification

config.json

@@ -1,79 +1,93 @@
 {
-  "_name_or_path": "ai-research-lab/bert-question-classifier",
+  "_name_or_path": "google-bert/bert-base-uncased",
   "architectures": [
     "BertForSequenceClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
-  "custom_pipelines": {
-    "question-classifier": {
-      "default": {
-        "model": {
-          "pt": [
-            "ai-research-lab/bert-question-classifier",
-            "main"
-          ]
-        }
-      },
-      "impl": "classifier_pipeline.MultiTaskClassifierPipeline",
-      "pt": [
-        "AutoModelForSequenceClassification"
-      ],
-      "tf": [],
-      "type": "text"
-    }
-  },
   "gradient_checkpointing": false,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
-  "hidden_size": 1024,
+  "hidden_size": 768,
   "id2label": {
     "0": "d",
     "1": "y",
     "2": "c",
     "3": "o",
-    "4": "career",
-    "5": "family",
-    "6": "marriage",
-    "7": "love",
-    "8": "self",
-    "9": "education",
-    "10": "wealth",
-    "11": "business",
-    "12": "hard times",
-    "13": "health",
-    "14": "spirituality",
-    "15": "others",
-    "16": "time_based",
-    "17": "positive_perception"
+    "4": "self",
+    "5": "health",
+    "6": "accumulated wealth",
+    "7": "family",
+    "8": "social media",
+    "9": "short travel",
+    "10": "sports",
+    "11": "property",
+    "12": "primary education",
+    "13": "love",
+    "14": "romance",
+    "15": "children",
+    "16": "higher education",
+    "17": "job",
+    "18": "diseases",
+    "19": "hard times",
+    "20": "competitive exam",
+    "21": "marriage",
+    "22": "business",
+    "23": "life span",
+    "24": "unearned wealth",
+    "25": "spirituality",
+    "26": "highest education",
+    "27": "long travel",
+    "28": "career",
+    "29": "income",
+    "30": "foreign",
+    "31": "expense",
+    "32": "time_based",
+    "33": "positive_perception"
   },
   "initializer_range": 0.02,
-  "intermediate_size": 4096,
+  "intermediate_size": 3072,
   "label2id": {
-    "business": 11,
+    "accumulated wealth": 6,
+    "business": 22,
     "c": 2,
-    "career": 4,
+    "career": 28,
+    "children": 15,
+    "competitive exam": 20,
     "d": 0,
-    "education": 9,
-    "family": 5,
-    "hard times": 12,
-    "health": 13,
-    "love": 7,
-    "marriage": 6,
+    "diseases": 18,
+    "expense": 31,
+    "family": 7,
+    "foreign": 30,
+    "hard times": 19,
+    "health": 5,
+    "higher education": 16,
+    "highest education": 26,
+    "income": 29,
+    "job": 17,
+    "life span": 23,
+    "long travel": 27,
+    "love": 13,
+    "marriage": 21,
     "o": 3,
-    "others": 15,
-    "positive_perception": 17,
-    "self": 8,
-    "spirituality": 14,
-    "time_based": 16,
-    "wealth": 10,
+    "positive_perception": 33,
+    "primary education": 12,
+    "property": 11,
+    "romance": 14,
+    "self": 4,
+    "short travel": 9,
+    "social media": 8,
+    "spirituality": 25,
+    "sports": 10,
+    "time_based": 32,
+    "unearned wealth": 24,
     "y": 1
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",
-  "num_attention_heads": 16,
-  "num_hidden_layers": 24,
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
   "task_specific_params": {
@@ -106,52 +120,116 @@
           "column": "category",
           "labels": [
             [
-              "career",
-              "Career"
+              "self",
+              "Self"
+            ],
+            [
+              "health",
+              "Health"
+            ],
+            [
+              "accumulated wealth",
+              "Accumulated Wealth"
             ],
             [
               "family",
               "Family"
             ],
             [
-              "marriage",
-              "Marriage"
+              "social media",
+              "Social media"
+            ],
+            [
+              "short travel",
+              "Short Travel"
+            ],
+            [
+              "sports",
+              "Sports"
+            ],
+            [
+              "property",
+              "Property"
+            ],
+            [
+              "primary education",
+              "Primary Education"
             ],
             [
               "love",
               "Love"
             ],
             [
-              "self",
-              "Self"
+              "romance",
+              "Romance"
             ],
             [
-              "education",
-              "Education"
+              "children",
+              "Children"
             ],
             [
-              "wealth",
-              "Wealth"
+              "higher education",
+              "Higher Education"
             ],
             [
-              "business",
-              "Business"
+              "job",
+              "Job"
+            ],
+            [
+              "diseases",
+              "Diseases"
             ],
             [
               "hard times",
               "Hard Times"
             ],
             [
-              "health",
-              "Health"
+              "competitive exam",
+              "Competitive Exam"
+            ],
+            [
+              "marriage",
+              "Marriage"
+            ],
+            [
+              "business",
+              "Business"
+            ],
+            [
+              "life span",
+              "Life Span"
+            ],
+            [
+              "unearned wealth",
+              "Unearned Wealth"
             ],
             [
               "spirituality",
               "Spirituality"
             ],
             [
-              "others",
-              "Others"
+              "highest education",
+              "Highest Education"
+            ],
+            [
+              "long travel",
+              "Long Travel"
+            ],
+            [
+              "career",
+              "Career"
+            ],
+            [
+              "income",
+              "Income"
+            ],
+            [
+              "foreign",
+              "Foreign"
+            ],
+            [
+              "expense",
+              "Expense"
             ]
           ],
           "loss_weight": 1,
@@ -190,7 +268,7 @@
     }
   },
   "torch_dtype": "float32",
-  "transformers_version": "4.47.1",
+  "transformers_version": "4.48.1",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522