Upload inference_config.json

inference_config.json

@@ -0,0 +1,44 @@
+{
+  "model_name": "mental/mental-bert-base-uncased",
+  "max_len": 128,
+  "n_classes": 8,
+  "classes": [
+    "Anxiety",
+    "Bipolar",
+    "Depression",
+    "Directed Aggression",
+    "Normal",
+    "Personality Disorder",
+    "Stress",
+    "Suicidal"
+  ],
+  "n_sources": 6,
+  "source_map": {
+    "cssrs": 0,
+    "huggingface": 1,
+    "kaggle": 2,
+    "kaggle_bpd": 3,
+    "olid": 4,
+    "swmh": 5
+  },
+  "source_reliability": {
+    "cssrs": 1.0,
+    "olid": 1.0,
+    "kaggle_bpd": 0.95,
+    "huggingface": 0.7,
+    "kaggle": 0.7,
+    "swmh": 0.5
+  },
+  "lambda_src": 0.3,
+  "best_epoch": 3,
+  "val_f1_macro": 0.8395463482978247,
+  "test_accuracy": 0.8322668327764879,
+  "test_f1_macro": 0.8380939220912206,
+  "test_f1_weighted": 0.8321028523440407,
+  "dep_to_sui": 715,
+  "sui_to_dep": 544,
+  "total_bleed": 1259,
+  "dataset_version": "v5",
+  "architecture": "source_aware_multitask_flat",
+  "hub_repo": "itsLu/mentalbert-v5-source-aware"
+}