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,811 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": 28,
  "format_version": 2,
  "model_name": "eegpt",
  "parameters": 25589507,
  "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",
  "sfreq": 250,
  "split_seed": 2026,
  "subtype_labels": [],
  "test_metrics": {
    "accuracy": 0.9201637666325486,
    "auprc": 0.65324809994628,
    "auroc": 0.8640698911466086,
    "balanced_accuracy": 0.767579570688379,
    "f1": 0.6138613861386139,
    "false_positives_per_minute": 0.5440414507772021,
    "fn": 186,
    "fp": 126,
    "n": 3908,
    "precision": 0.6631016042780749,
    "prevalence": 0.11105424769703173,
    "sensitivity": 0.5714285714285714,
    "specificity": 0.9637305699481865,
    "threshold": 0.8916015625,
    "tn": 3348,
    "tp": 248
  },
  "trainable_probe_parameters": 302339,
  "training_strategy": "pretrained_encoder_frozen_ied_temporal_head",
  "validation_metrics": {
    "accuracy": 0.9420215495125706,
    "auprc": 0.5628674715868012,
    "auroc": 0.8119094063849486,
    "balanced_accuracy": 0.7289665627716642,
    "f1": 0.5703422053231939,
    "false_positives_per_minute": 0.25125628140703515,
    "fn": 166,
    "fp": 60,
    "n": 3898,
    "precision": 0.7142857142857143,
    "prevalence": 0.08106721395587481,
    "sensitivity": 0.47468354430379744,
    "specificity": 0.983249581239531,
    "threshold": 0.8916015625,
    "tn": 3522,
    "tp": 150
  },
  "validation_threshold_f1": 0.8916015625,
  "validation_threshold_sensitivity80": 0.007965087890625,
  "window_samples": 1000,
  "window_seconds": 4.0
}