auris-models / deep_learning_results.json
Rthur2003's picture
Upload deep_learning_results.json with huggingface_hub
aeb7a07 verified
Raw History Blame Contribute Delete
1.48 kB
{
"Deep MLP (512-256-128-64)": {
"accuracy": 0.8849,
"precision": 0.841,
"recall": 0.879,
"f1": 0.8596,
"roc_auc": 0.9537,
"fold_aucs": [
0.9582,
0.9557,
0.9508,
0.9492,
0.9571
],
"train_time_sec": 81.4,
"type": "deep_learning",
"model_path": "models\\model_dl_deep_mlp_512_256_128_64.pkl"
},
"1D-CNN": {
"accuracy": 0.7665,
"precision": 0.6987,
"recall": 0.7339,
"f1": 0.7159,
"roc_auc": 0.8442,
"fold_aucs": [
0.8583,
0.8645,
0.8589,
0.8394,
0.8502
],
"train_time_sec": 125.0,
"type": "deep_learning",
"model_path": "models\\model_dl_1d_cnn.pkl"
},
"Residual MLP (3 blocks)": {
"accuracy": 0.8756,
"precision": 0.833,
"recall": 0.8626,
"f1": 0.8476,
"roc_auc": 0.9453,
"fold_aucs": [
0.9523,
0.9491,
0.9473,
0.9407,
0.9531
],
"train_time_sec": 128.7,
"type": "deep_learning",
"model_path": "models\\model_dl_residual_mlp_3_blocks.pkl"
},
"Attention MLP": {
"accuracy": 0.8628,
"precision": 0.8267,
"recall": 0.8319,
"f1": 0.8293,
"roc_auc": 0.9356,
"fold_aucs": [
0.9318,
0.9461,
0.9379,
0.932,
0.9316
],
"train_time_sec": 149.6,
"type": "deep_learning",
"model_path": "models\\model_dl_attention_mlp.pkl"
}
}