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"
  }
}