{
  "architecture": "V5 Flat 8-Class MentalBERT",
  "dataset": "mohamedasem318/mental-health-dataset-extended-v5",
  "backbone": "mental/mental-bert-base-uncased",
  "num_classes": 8,
  "classes": [
    "Anxiety",
    "Bipolar",
    "Depression",
    "Directed Aggression",
    "Normal",
    "Personality Disorder",
    "Stress",
    "Suicidal"
  ],
  "id2label": {
    "0": "Anxiety",
    "1": "Bipolar",
    "2": "Depression",
    "3": "Directed Aggression",
    "4": "Normal",
    "5": "Personality Disorder",
    "6": "Stress",
    "7": "Suicidal"
  },
  "label2id": {
    "Anxiety": 0,
    "Bipolar": 1,
    "Depression": 2,
    "Directed Aggression": 3,
    "Normal": 4,
    "Personality Disorder": 5,
    "Stress": 6,
    "Suicidal": 7
  },
  "max_len": 128,
  "batch_size": 32,
  "learning_rate": 2e-05,
  "epochs": 4,
  "label_smoothing": 0.05,
  "imbalance": "WeightedRandomSampler + class-weighted CE (capped at 3.0)",
  "best_val_f1": 0.8307269417033605,
  "test_metrics": {
    "accuracy": 0.8283594767540631,
    "f1_macro": 0.8350278478225386,
    "f1_weighted": 0.8280225059714043,
    "dep_to_sui": 733,
    "sui_to_dep": 516,
    "total_bleed": 1249,
    "roc_auc_macro": 0.9638191731454191
  },
  "baselines": {
    "mb_flat_v3_deployed": {
      "acc": 0.8198,
      "f1": 0.7646,
      "bleed": 1224
    },
    "mb_flat_v5_ph10": {
      "acc": 0.8241,
      "f1": 0.8308,
      "bleed": 1265
    },
    "v5_3stage_flat_lf": {
      "acc": 0.8404,
      "f1": 0.846,
      "bleed": 1179,
      "sui_to_dep": 274
    }
  }
}