Hugging Face's logo

SenuaLab
/
EEG-IED-Detection

PyTorch
Safetensors
eeg
brain-signal
biosignal
time-series
classification
biomedical-signal-processing
epilepsy
interictal-epileptiform-discharge
neuroscience
eegpt
eeg-dino
braindecode
Model card Files
xet
 Community
EEG-IED-Detection
File size: 2,824 Bytes
dc9d5f6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
{
  "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
}