inference_config.json

{
  "architecture": "cooperative_source_aware_cleanlab_v1",
  "base_model": "mental/mental-bert-base-uncased",
  "dataset": "mohamedasem318/mental-health-dataset-extended-v5",
  "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
  },
  "max_len": 128,
  "epochs": 4,
  "batch_size": 32,
  "lr": 2e-05,
  "warmup_ratio": 0.1,
  "src_loss_weight": 0.3,
  "cleanlab_drops": 4745,
  "cleanlab_drops_by_source": {
    "cssrs": 0,
    "huggingface": 683,
    "kaggle": 4062,
    "kaggle_bpd": 0,
    "olid": 0,
    "swmh": 0
  },
  "cleanlab_scope": "depression_suicidal_non_cssrs",
  "cleanlab_version": "2.6.x",
  "kfold_n_splits": 3,
  "kfold_oof_epochs": 2,
  "best_epoch": 4,
  "val_f1_macro": 0.8693227702265434,
  "test_accuracy": 0.8314174075542217,
  "test_f1_macro": 0.8384363019984747,
  "test_f1_weighted": 0.8313469552418437,
  "test_bleed_dep_sui": 692,
  "test_bleed_sui_dep": 575,
  "test_bleed_total": 1267,
  "inference_note": "Load bert backbone via BertModel.from_pretrained. Reattach cls_head (Dropout(0.1) + Linear(768, n_classes)) from cls_head.pt. src_head is training-only and not needed for inference."
}