{
  "architecture": "NEDAClassifier",
  "backbone": "mental/mental-roberta-base",
  "num_labels": 5,
  "label2id": {
    "Anxious": 0,
    "Depressed": 1,
    "Frustrated": 2,
    "Others": 3,
    "Suicidal": 4
  },
  "id2label": {
    "0": "Anxious",
    "1": "Depressed",
    "2": "Frustrated",
    "3": "Others",
    "4": "Suicidal"
  },
  "max_length": 256,
  "neda_config": {
    "model_name": "mental/mental-roberta-base",
    "max_length": 256,
    "hidden_dim": 512,
    "dropout": 0.1,
    "use_nacm": true,
    "use_clgaf": true,
    "use_mgp": true,
    "use_supcon": true,
    "use_fgm": true,
    "use_ema": true,
    "neg_scope_max": 10,
    "neg_context_window": 3,
    "num_fuse_layers": 4,
    "fuse_heads": 4,
    "num_pseudo_sents": 4,
    "supcon_dim": 128,
    "supcon_alpha": 0.15,
    "supcon_temp": 0.07,
    "fgm_epsilon": 0.5,
    "ema_decay": 0.999,
    "neg_aux_weight": 0.05,
    "epochs": 30,
    "batch_size": 64,
    "encoder_lr": 2e-05,
    "head_lr": 0.0001,
    "weight_decay": 0.01,
    "warmup_ratio": 0.1,
    "patience": 5,
    "max_grad_norm": 1.0,
    "n_folds": 3,
    "seed": 42,
    "data_path": "/kaggle/input/datasets/sumitnath07/mental-distress/MentalDistress_Dataset_clean.csv",
    "output_dir": "/kaggle/working/neda_checkpoints"
  },
  "metrics": {
    "accuracy": 0.8629320619785459,
    "f1_weighted": 0.8623690312008874,
    "f1_macro": 0.8651399633181327
  }
}