{
  "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"
}