pytorch/eegpt_linear_probe/config.json

{
  "binary_labels": [
    "Non-IED",
    "IED"
  ],
  "channels": [
    "Fp1",
    "Fp2",
    "F3",
    "F4",
    "C3",
    "C4",
    "P3",
    "P4",
    "O1",
    "O2",
    "F7",
    "F8",
    "T3",
    "T4",
    "T5",
    "T6",
    "Fz",
    "Cz",
    "Pz"
  ],
  "eegpt_channel_names": [
    "Fp1",
    "Fp2",
    "F3",
    "F4",
    "C3",
    "C4",
    "P3",
    "P4",
    "O1",
    "O2",
    "F7",
    "F8",
    "T7",
    "T8",
    "P7",
    "P8",
    "Fz",
    "Cz",
    "Pz"
  ],
  "epoch": 19,
  "format_version": 2,
  "model_name": "eegpt_linear",
  "parameters": 25320946,
  "preprocessing": {
    "bandpass_hz": [
      1.0,
      45.0
    ],
    "normalization": "global_window_zscore_clip8",
    "reference": "average"
  },
  "pretrained": {
    "intentionally_not_loaded": [
      "chans_id (downstream 19-channel identity buffer)",
      "final_layer.* (new binary IED classification head)"
    ],
    "loaded_tensors": 102,
    "loaded_values": 25287168,
    "missing_after_load": [
      "chans_id"
    ],
    "repo_id": "braindecode/eegpt-pretrained",
    "resolved_revision": "e41cb3ae2ce4fd9eb736862292c91f8128d15618",
    "unexpected_after_load": [],
    "weights_file": "model.safetensors",
    "weights_sha256": "fb34c20609983324679b9534f9d17a2289a232d0276dd2b8b7d5b088f876f621"
  },
  "release_name": "eegpt_linear_probe",
  "sfreq": 250,
  "split_seed": 2026,
  "subtype_labels": [],
  "test_metrics": {
    "accuracy": 0.7131525076765609,
    "auprc": 0.3722106087915067,
    "auroc": 0.7519221126525154,
    "balanced_accuracy": 0.6702979871540794,
    "f1": 0.32265861027190335,
    "false_positives_per_minute": 4.119170984455959,
    "fn": 167,
    "fp": 954,
    "n": 3908,
    "precision": 0.21867321867321868,
    "prevalence": 0.11105424769703173,
    "sensitivity": 0.6152073732718893,
    "specificity": 0.7253886010362695,
    "threshold": 0.40234375,
    "tn": 2520,
    "tp": 267
  },
  "trainable_probe_parameters": 33778,
  "training_strategy": "pretrained_encoder_frozen_linear_probe",
  "validation_metrics": {
    "accuracy": 0.8496664956387892,
    "auprc": 0.23998456130581225,
    "auroc": 0.7217963940659697,
    "balanced_accuracy": 0.6715018482002135,
    "f1": 0.3310502283105023,
    "false_positives_per_minute": 1.7378559463986598,
    "fn": 171,
    "fp": 415,
    "n": 3898,
    "precision": 0.25892857142857145,
    "prevalence": 0.08106721395587481,
    "sensitivity": 0.4588607594936709,
    "specificity": 0.884142936906756,
    "threshold": 0.40234375,
    "tn": 3167,
    "tp": 145
  },
  "validation_threshold_f1": 0.40234375,
  "validation_threshold_sensitivity80": 0.049224853515625,
  "window_samples": 1000,
  "window_seconds": 4.0
}