{
  "config": {
    "checkpoint": "checkpoints/S_variable64_v6_r3_20260329_160409/best_model.pt",
    "output_dir": "eval/wellness_finetune_dann_lambda03",
    "device": "cuda",
    "epochs": 20,
    "batch_size": 8,
    "lr": 2e-05,
    "head_lr": 0.001,
    "cohort_head_lr": 0.001,
    "lambda_max": 0.3,
    "lambda_warmup_epochs": 3,
    "weight_decay": 0.0001,
    "chunk_seconds": 30,
    "patience": 5,
    "train_last_n_blocks": 2,
    "seed": 42,
    "num_workers": 4,
    "pool": "+sub+reg",
    "val_frac": 0.15
  },
  "best_epoch": 14,
  "best_val_auc": 0.9581392640894271,
  "patient_overlap_audit": 0,
  "test": {
    "auc": 0.5891215768028049,
    "brier": 0.3347329579899468,
    "ece": 0.2991181795250857,
    "mce": 0.42774906348021785,
    "n": 356,
    "reliability": {
      "mean_pred": [
        0.01669538096422657,
        0.03363641209647942,
        0.10694517404462561,
        0.26541023876531766,
        0.4257866898143904,
        0.5664715403594813,
        0.71542028606602,
        0.9324916355589318,
        0.999138844678747,
        0.9997537109148897
      ],
      "frac_pos": [
        0.4444444444444444,
        0.4166666666666667,
        0.5,
        0.4166666666666667,
        0.6111111111111112,
        0.3333333333333333,
        0.5142857142857142,
        0.5714285714285714,
        0.6285714285714286,
        0.7142857142857143
      ],
      "weights": [
        0.10112359550561797,
        0.10112359550561797,
        0.10112359550561797,
        0.10112359550561797,
        0.10112359550561797,
        0.10112359550561797,
        0.09831460674157304,
        0.09831460674157304,
        0.09831460674157304,
        0.09831460674157304
      ]
    }
  },
  "train_log": [
    {
      "epoch": 1,
      "lambda": 0.09999999999999999,
      "well_loss": 0.16510153805815375,
      "cohort_loss": 0.11694759285957716,
      "cohort_acc": 0.9622093023255814,
      "val_auc": 0.9334536562645552,
      "val_brier": 0.09610738803213281
    },
    {
      "epoch": 2,
      "lambda": 0.19999999999999998,
      "well_loss": 0.14242211364610338,
      "cohort_loss": 0.05007892586274497,
      "cohort_acc": 0.9767441860465116,
      "val_auc": 0.9419538891476478,
      "val_brier": 0.09814545826650627
    },
    {
      "epoch": 3,
      "lambda": 0.3,
      "well_loss": 0.14403539020026665,
      "cohort_loss": 0.042790586666810675,
      "cohort_acc": 0.9825581395348837,
      "val_auc": 0.9487657196087564,
      "val_brier": 0.09255905836683599
    },
    {
      "epoch": 4,
      "lambda": 0.3,
      "well_loss": 0.14312272210237245,
      "cohort_loss": 0.0359738634272492,
      "cohort_acc": 0.9796511627906976,
      "val_auc": 0.9465533302282254,
      "val_brier": 0.10412288481990085
    },
    {
      "epoch": 5,
      "lambda": 0.3,
      "well_loss": 0.140509495074146,
      "cohort_loss": 0.04051422904929096,
      "cohort_acc": 0.9813953488372092,
      "val_auc": 0.9461457848160223,
      "val_brier": 0.11304157247617672
    },
    {
      "epoch": 6,
      "lambda": 0.3,
      "well_loss": 0.13898292016183827,
      "cohort_loss": 0.038494305121514125,
      "cohort_acc": 0.9784883720930233,
      "val_auc": 0.952433628318584,
      "val_brier": 0.08156782784187082
    },
    {
      "epoch": 7,
      "lambda": 0.3,
      "well_loss": 0.13735196732504423,
      "cohort_loss": 0.034704251026434006,
      "cohort_acc": 0.9837209302325581,
      "val_auc": 0.9542966930600839,
      "val_brier": 0.08827166573269962
    },
    {
      "epoch": 8,
      "lambda": 0.3,
      "well_loss": 0.13313717285799254,
      "cohort_loss": 0.032732733793972034,
      "cohort_acc": 0.9790697674418605,
      "val_auc": 0.9512692128551467,
      "val_brier": 0.08459728707706181
    },
    {
      "epoch": 9,
      "lambda": 0.3,
      "well_loss": 0.1310320912344587,
      "cohort_loss": 0.028951282990718415,
      "cohort_acc": 0.9784883720930233,
      "val_auc": 0.95418025151374,
      "val_brier": 0.09433399210026669
    },
    {
      "epoch": 10,
      "lambda": 0.3,
      "well_loss": 0.1305306267112407,
      "cohort_loss": 0.030289230750013222,
      "cohort_acc": 0.9813953488372092,
      "val_auc": 0.9541802515137402,
      "val_brier": 0.08075600936280382
    },
    {
      "epoch": 11,
      "lambda": 0.3,
      "well_loss": 0.13125988908117464,
      "cohort_loss": 0.028573764901692424,
      "cohort_acc": 0.9837209302325581,
      "val_auc": 0.9571495109455054,
      "val_brier": 0.07474660777411347
    },
    {
      "epoch": 12,
      "lambda": 0.3,
      "well_loss": 0.12154618137391544,
      "cohort_loss": 0.030648156544316425,
      "cohort_acc": 0.9796511627906976,
      "val_auc": 0.9554028877503493,
      "val_brier": 0.08075307574395674
    },
    {
      "epoch": 13,
      "lambda": 0.3,
      "well_loss": 0.12668655127839207,
      "cohort_loss": 0.02899676738899242,
      "cohort_acc": 0.9825581395348837,
      "val_auc": 0.9576152771308802,
      "val_brier": 0.08534399510630883
    },
    {
      "epoch": 14,
      "lambda": 0.3,
      "well_loss": 0.12221924414490025,
      "cohort_loss": 0.026997907841683308,
      "cohort_acc": 0.9808139534883721,
      "val_auc": 0.9581392640894271,
      "val_brier": 0.08502427335589301
    },
    {
      "epoch": 15,
      "lambda": 0.3,
      "well_loss": 0.12406214804843414,
      "cohort_loss": 0.02808780605919719,
      "cohort_acc": 0.9843023255813953,
      "val_auc": 0.9567419655333023,
      "val_brier": 0.07816887862596124
    },
    {
      "epoch": 16,
      "lambda": 0.3,
      "well_loss": 0.1162589408137814,
      "cohort_loss": 0.0277483245660242,
      "cohort_acc": 0.9819767441860465,
      "val_auc": 0.9580810433162552,
      "val_brier": 0.08149202462604734
    },
    {
      "epoch": 17,
      "lambda": 0.3,
      "well_loss": 0.12114277525025224,
      "cohort_loss": 0.029295074426359986,
      "cohort_acc": 0.9808139534883721,
      "val_auc": 0.9580228225430834,
      "val_brier": 0.08029980881215418
    },
    {
      "epoch": 18,
      "lambda": 0.3,
      "well_loss": 0.11439366689896709,
      "cohort_loss": 0.028150552303595377,
      "cohort_acc": 0.9808139534883721,
      "val_auc": 0.9578481602235678,
      "val_brier": 0.08095781078244582
    },
    {
      "epoch": 19,
      "lambda": 0.3,
      "well_loss": 0.11256319967634586,
      "cohort_loss": 0.026544157259170368,
      "cohort_acc": 0.9848837209302326,
      "val_auc": 0.9577899394503959,
      "val_brier": 0.08087899166886207
    }
  ]
}