{
  "checkpoint": "checkpoints/pretrain_nextpatch_v7_with_readtbi_20260130_102320/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.9181818181818182,
      "balanced_accuracy": 0.9178947368421053,
      "f1": 0.9274193548387096,
      "auc": 0.9706105263157894,
      "confusion_matrix": [
        [
          87,
          8
        ],
        [
          10,
          115
        ]
      ]
    },
    "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.5439560439560439,
      "balanced_accuracy": 0.5520731707317073,
      "f1": 0.5310734463276836,
      "auc": 0.6501219512195122,
      "confusion_matrix": [
        [
          52,
          30
        ],
        [
          53,
          47
        ]
      ]
    },
    "parkinson": {
      "task": "parkinson",
      "n_train": 1359,
      "n_test": 135,
      "n_pos_test": 77,
      "n_neg_test": 58,
      "accuracy": 0.8,
      "balanced_accuracy": 0.8225481415136587,
      "f1": 0.7906976744186046,
      "auc": 0.822884012539185,
      "confusion_matrix": [
        [
          57,
          1
        ],
        [
          26,
          51
        ]
      ]
    },
    "schizophrenia": {
      "task": "schizophrenia",
      "n_train": 102,
      "n_test": 22,
      "n_pos_test": 9,
      "n_neg_test": 13,
      "accuracy": 0.9090909090909091,
      "balanced_accuracy": 0.9230769230769231,
      "f1": 0.9,
      "auc": 0.8461538461538461,
      "confusion_matrix": [
        [
          11,
          2
        ],
        [
          0,
          9
        ]
      ]
    },
    "tbi": {
      "task": "tbi",
      "n_train": 407,
      "n_test": 94,
      "n_pos_test": 33,
      "n_neg_test": 61,
      "accuracy": 1.0,
      "balanced_accuracy": 1.0,
      "f1": 1.0,
      "auc": 1.0,
      "confusion_matrix": [
        [
          61,
          0
        ],
        [
          0,
          33
        ]
      ]
    }
  },
  "mean_auc": 0.8816283893713887
}