aryakoureshi's picture
Release SenuaLab patient-independent EEG IED models v1.0.0
dc9d5f6 verified
Raw
History Blame Contribute Delete
3.17 kB
{
"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"
],
"eegdino_channel_names": [
"Fp1",
"Fp2",
"F3",
"F4",
"C3",
"C4",
"P3",
"P4",
"O1",
"O2",
"F7",
"F8",
"T3",
"T4",
"T5",
"T6",
"Fz",
"Cz",
"Pz"
],
"epoch": 11,
"format_version": 2,
"model_name": "eegdino_medium",
"parameters": 34445250,
"preprocessing": {
"bandpass_hz": [
1.0,
45.0
],
"normalization": "global_window_zscore_clip8",
"reference": "average"
},
"pretrained": {
"intentionally_not_loaded": [
"final_layer.* (new downstream binary IED head)"
],
"loaded_tensors": 227,
"loaded_values": 33755456,
"missing_after_load": [
"final_layer.token_proj.0.weight",
"final_layer.token_proj.0.bias",
"final_layer.time_proj.0.weight",
"final_layer.time_proj.0.bias",
"final_layer.classifier.0.weight",
"final_layer.classifier.0.bias",
"final_layer.classifier.3.weight",
"final_layer.classifier.3.bias",
"final_layer.classifier.6.weight",
"final_layer.classifier.6.bias"
],
"repo_id": "braindecode/eegdino-medium-pretrained",
"requested_revision": "191f73eb50d68a184b4cabb623938df914235d3c",
"resolved_revision": "191f73eb50d68a184b4cabb623938df914235d3c",
"unexpected_after_load": [],
"weights_file": "model.safetensors",
"weights_sha256": "17a73d056afde262d1be0cc21434ca751855691726a0500b3f3c809e2eecdf9e"
},
"release_name": "eegdino_medium",
"sfreq": 200,
"split_seed": 2026,
"subtype_labels": [],
"test_metrics": {
"accuracy": 0.9398669396110543,
"auprc": 0.8019567240287226,
"auroc": 0.9362860777493904,
"balanced_accuracy": 0.8683870171836076,
"f1": 0.7414741474147415,
"false_positives_per_minute": 0.5958549222797928,
"fn": 97,
"fp": 138,
"n": 3908,
"precision": 0.7094736842105264,
"prevalence": 0.11105424769703173,
"sensitivity": 0.7764976958525346,
"specificity": 0.9602763385146805,
"threshold": 0.619140625,
"tn": 3336,
"tp": 337
},
"training_strategy": "partial_last_2",
"validation_metrics": {
"accuracy": 0.9461262185736276,
"auprc": 0.6916968240811774,
"auroc": 0.8983352946165426,
"balanced_accuracy": 0.7903503099180855,
"f1": 0.6452702702702703,
"false_positives_per_minute": 0.35594639865996647,
"fn": 125,
"fp": 85,
"n": 3898,
"precision": 0.6920289855072463,
"prevalence": 0.08106721395587481,
"sensitivity": 0.6044303797468354,
"specificity": 0.9762702400893356,
"threshold": 0.619140625,
"tn": 3497,
"tp": 191
},
"validation_threshold_f1": 0.619140625,
"validation_threshold_sensitivity80": 0.056121826171875,
"window_samples": 800,
"window_seconds": 4.0
}