File size: 1,476 Bytes
aeb7a07 | 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 | {
"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"
}
} |