EEG-IED-Detection / ensembles /ensemble_4model.json
aryakoureshi's picture
Release SenuaLab patient-independent EEG IED models v1.0.0
dc9d5f6 verified
Raw
History Blame Contribute Delete
1.39 kB
{
"aggregation": "equal_probability_mean",
"models": [
"iednet_lite",
"resnet_attention",
"eegpt",
"eegdino_medium"
],
"selection_score_validation_auprc_plus_025_f1": 0.8335915544686525,
"test_metrics": {
"accuracy": 0.9483111566018424,
"auprc": 0.7888269120699914,
"auroc": 0.9200864088462283,
"balanced_accuracy": 0.8600306689058151,
"f1": 0.7623529411764706,
"false_positives_per_minute": 0.39723661485319517,
"fn": 110,
"fp": 92,
"n": 3908,
"precision": 0.7788461538461539,
"prevalence": 0.11105424769703173,
"sensitivity": 0.7465437788018433,
"specificity": 0.973517559009787,
"threshold": 0.5273836475,
"tn": 3382,
"tp": 324
},
"threshold": 0.5273836475,
"validation_metrics": {
"accuracy": 0.9512570548999487,
"auprc": 0.6725428653300758,
"auroc": 0.8953611234795638,
"balanced_accuracy": 0.7657309048759975,
"f1": 0.6441947565543071,
"false_positives_per_minute": 0.19262981574539362,
"fn": 144,
"fp": 46,
"n": 3898,
"precision": 0.7889908256880734,
"prevalence": 0.08106721395587481,
"sensitivity": 0.5443037974683544,
"specificity": 0.9871580122836404,
"threshold": 0.5273836475,
"tn": 3536,
"tp": 172
},
"weights": [
0.25,
0.25,
0.25,
0.25
]
}