EEG-IED-Detection / ensembles /ensemble_eegpt.json
aryakoureshi's picture
Release SenuaLab patient-independent EEG IED models v1.0.0
dc9d5f6 verified
Raw
History Blame Contribute Delete
1.33 kB
{
"aggregation": "equal_probability_mean",
"models": [
"iednet_lite",
"resnet_attention",
"eegpt"
],
"test_metrics": {
"accuracy": 0.936028659160696,
"auprc": 0.7451331821970812,
"auroc": 0.9004447787249058,
"balanced_accuracy": 0.8248940781950977,
"f1": 0.7030878859857482,
"false_positives_per_minute": 0.4835924006908463,
"fn": 138,
"fp": 112,
"n": 3908,
"precision": 0.7254901960784313,
"prevalence": 0.11105424769703173,
"sensitivity": 0.6820276497695853,
"specificity": 0.9677605066206102,
"threshold": 0.5290762755,
"tn": 3362,
"tp": 296
},
"threshold": 0.5290762755,
"validation_metrics": {
"accuracy": 0.9499743458183684,
"auprc": 0.6261332028552472,
"auroc": 0.8396483118829026,
"balanced_accuracy": 0.7578195124709342,
"f1": 0.6313799621928167,
"false_positives_per_minute": 0.19262981574539362,
"fn": 149,
"fp": 46,
"n": 3898,
"precision": 0.784037558685446,
"prevalence": 0.08106721395587481,
"sensitivity": 0.5284810126582279,
"specificity": 0.9871580122836404,
"threshold": 0.5290762755,
"tn": 3536,
"tp": 167
},
"weights": [
0.3333333333333333,
0.3333333333333333,
0.3333333333333333
]
}