Rthur2003 commited on
Commit
aeb7a07
·
verified ·
1 Parent(s): 725537c

Upload deep_learning_results.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. deep_learning_results.json +70 -0
deep_learning_results.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "Deep MLP (512-256-128-64)": {
3
+ "accuracy": 0.8849,
4
+ "precision": 0.841,
5
+ "recall": 0.879,
6
+ "f1": 0.8596,
7
+ "roc_auc": 0.9537,
8
+ "fold_aucs": [
9
+ 0.9582,
10
+ 0.9557,
11
+ 0.9508,
12
+ 0.9492,
13
+ 0.9571
14
+ ],
15
+ "train_time_sec": 81.4,
16
+ "type": "deep_learning",
17
+ "model_path": "models\\model_dl_deep_mlp_512_256_128_64.pkl"
18
+ },
19
+ "1D-CNN": {
20
+ "accuracy": 0.7665,
21
+ "precision": 0.6987,
22
+ "recall": 0.7339,
23
+ "f1": 0.7159,
24
+ "roc_auc": 0.8442,
25
+ "fold_aucs": [
26
+ 0.8583,
27
+ 0.8645,
28
+ 0.8589,
29
+ 0.8394,
30
+ 0.8502
31
+ ],
32
+ "train_time_sec": 125.0,
33
+ "type": "deep_learning",
34
+ "model_path": "models\\model_dl_1d_cnn.pkl"
35
+ },
36
+ "Residual MLP (3 blocks)": {
37
+ "accuracy": 0.8756,
38
+ "precision": 0.833,
39
+ "recall": 0.8626,
40
+ "f1": 0.8476,
41
+ "roc_auc": 0.9453,
42
+ "fold_aucs": [
43
+ 0.9523,
44
+ 0.9491,
45
+ 0.9473,
46
+ 0.9407,
47
+ 0.9531
48
+ ],
49
+ "train_time_sec": 128.7,
50
+ "type": "deep_learning",
51
+ "model_path": "models\\model_dl_residual_mlp_3_blocks.pkl"
52
+ },
53
+ "Attention MLP": {
54
+ "accuracy": 0.8628,
55
+ "precision": 0.8267,
56
+ "recall": 0.8319,
57
+ "f1": 0.8293,
58
+ "roc_auc": 0.9356,
59
+ "fold_aucs": [
60
+ 0.9318,
61
+ 0.9461,
62
+ 0.9379,
63
+ 0.932,
64
+ 0.9316
65
+ ],
66
+ "train_time_sec": 149.6,
67
+ "type": "deep_learning",
68
+ "model_path": "models\\model_dl_attention_mlp.pkl"
69
+ }
70
+ }