{
  "checkpoint": "checkpoints/pretrain_nextpatch_v6_variable_20260127_203129_20260127_203136/best_model.pt",
  "config": {
    "n_channels": 21,
    "max_channels": 21,
    "hidden_dim": 768,
    "patch_size": 32,
    "n_layers": 6,
    "n_heads": 12,
    "use_spatial_rope": false
  },
  "results": {
    "alzheimer": {
      "task": "alzheimer",
      "n_train": 2249,
      "n_test": 220,
      "n_pos_test": 125,
      "n_neg_test": 95,
      "accuracy": 0.8818181818181818,
      "balanced_accuracy": 0.8833684210526316,
      "f1": 0.8934426229508197,
      "auc": 0.9478736842105262,
      "confusion_matrix": [
        [
          85,
          10
        ],
        [
          16,
          109
        ]
      ]
    },
    "ftd": {
      "task": "ftd",
      "n_train": 1563,
      "n_test": 70,
      "n_pos_test": 38,
      "n_neg_test": 32,
      "accuracy": 0.9857142857142858,
      "balanced_accuracy": 0.986842105263158,
      "f1": 0.9866666666666667,
      "auc": 1.0,
      "confusion_matrix": [
        [
          32,
          0
        ],
        [
          1,
          37
        ]
      ]
    },
    "mci": {
      "task": "mci",
      "n_train": 1874,
      "n_test": 182,
      "n_pos_test": 100,
      "n_neg_test": 82,
      "accuracy": 0.554945054945055,
      "balanced_accuracy": 0.5653658536585366,
      "f1": 0.5317919075144508,
      "auc": 0.6445121951219512,
      "confusion_matrix": [
        [
          55,
          27
        ],
        [
          54,
          46
        ]
      ]
    },
    "parkinson": {
      "task": "parkinson",
      "n_train": 1359,
      "n_test": 135,
      "n_pos_test": 77,
      "n_neg_test": 58,
      "accuracy": 0.7851851851851852,
      "balanced_accuracy": 0.8010523958799821,
      "f1": 0.7851851851851852,
      "auc": 0.8918495297805642,
      "confusion_matrix": [
        [
          53,
          5
        ],
        [
          24,
          53
        ]
      ]
    },
    "schizophrenia": {
      "task": "schizophrenia",
      "n_train": 102,
      "n_test": 22,
      "n_pos_test": 9,
      "n_neg_test": 13,
      "accuracy": 0.9545454545454546,
      "balanced_accuracy": 0.9615384615384616,
      "f1": 0.9473684210526315,
      "auc": 0.9914529914529915,
      "confusion_matrix": [
        [
          12,
          1
        ],
        [
          0,
          9
        ]
      ]
    },
    "tbi_within_readtbi": {
      "task": "tbi_within_readtbi",
      "n_train": 133,
      "n_test": 44,
      "n_pos_test": 23,
      "n_neg_test": 21,
      "accuracy": 0.5909090909090909,
      "balanced_accuracy": 0.5921325051759834,
      "f1": 0.5909090909090909,
      "auc": 0.5983436853002071,
      "confusion_matrix": [
        [
          13,
          8
        ],
        [
          10,
          13
        ]
      ]
    }
  },
  "mean_auc": 0.8456720143110399
}