Hugging Face's logo

Neural-Network-Project
/
ECG-Disease-Classifier

Keras
English
tensorflow
ecg
cardiology
medical
pediatric
time-series
multi-label-classification
cnn
Model card Files
xet
Community
kiril-buga commited on
Commit
e72c065
·
verified ·
1 Parent(s): 89e98f8

Add model metadata

Browse files
Files changed (1) hide show
  1. model_results.json +262 -0
model_results.json ADDED
@@ -0,0 +1,262 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "1D_CNN_19Classes_multilabel_19v3",
3
+ "version": "v03_balanced",
4
+ "created": "2025-12-07T14:07:16.688070",
5
+ "num_classes": 19,
6
+ "disease_names": [
7
+ "Fulminant/Viral Myocarditis",
8
+ "Acute Myocarditis",
9
+ "Myocarditis Unspecified",
10
+ "Dilated Cardiomyopathy",
11
+ "Hypertrophic Cardiomyopathy",
12
+ "Cardiomyopathy Unspecified",
13
+ "Noncompaction Ventricular Myocardium",
14
+ "Kawasaki Disease",
15
+ "Ventricular Septal Defect",
16
+ "Atrial Septal Defect",
17
+ "Atrioventricular Septal Defect",
18
+ "Tetralogy of Fallot",
19
+ "Pulmonary Valve Stenosis",
20
+ "Patent Ductus Arteriosus",
21
+ "Pulmonary Artery Stenosis",
22
+ "Pulmonary Valve Regurgitation",
23
+ "Mitral Valve Insufficiency",
24
+ "Congenital Heart Malformation",
25
+ "Healthy"
26
+ ],
27
+ "icd_codes": [
28
+ "I40.0",
29
+ "I40.9",
30
+ "I51.4",
31
+ "I42.0",
32
+ "I42.2",
33
+ "I42.9",
34
+ "Q24.8",
35
+ "M30.3",
36
+ "Q21.0",
37
+ "Q21.1",
38
+ "Q21.2",
39
+ "Q21.3",
40
+ "Q22.1",
41
+ "Q25.0",
42
+ "Q25.6",
43
+ "I37.0",
44
+ "I34.0",
45
+ "Q24.9",
46
+ "Healthy"
47
+ ],
48
+ "icd_to_disease_name": {
49
+ "I40.0": "Fulminant/Viral Myocarditis",
50
+ "I40.9": "Acute Myocarditis",
51
+ "I51.4": "Myocarditis Unspecified",
52
+ "I42.0": "Dilated Cardiomyopathy",
53
+ "I42.2": "Hypertrophic Cardiomyopathy",
54
+ "I42.9": "Cardiomyopathy Unspecified",
55
+ "Q24.8": "Noncompaction Ventricular Myocardium",
56
+ "M30.3": "Kawasaki Disease",
57
+ "Q21.0": "Ventricular Septal Defect",
58
+ "Q21.1": "Atrial Septal Defect",
59
+ "Q21.2": "Atrioventricular Septal Defect",
60
+ "Q21.3": "Tetralogy of Fallot",
61
+ "Q22.1": "Pulmonary Valve Stenosis",
62
+ "Q25.0": "Patent Ductus Arteriosus",
63
+ "Q25.6": "Pulmonary Artery Stenosis",
64
+ "I37.0": "Pulmonary Valve Regurgitation",
65
+ "I34.0": "Mitral Valve Insufficiency",
66
+ "Q24.9": "Congenital Heart Malformation",
67
+ "Healthy": "Healthy"
68
+ },
69
+ "disease_groups": {
70
+ "Myocarditis": [
71
+ "I40.0",
72
+ "I40.9",
73
+ "I51.4"
74
+ ],
75
+ "Cardiomyopathy": [
76
+ "I42.0",
77
+ "I42.2",
78
+ "I42.9",
79
+ "Q24.8"
80
+ ],
81
+ "Kawasaki": [
82
+ "M30.3"
83
+ ],
84
+ "CHD": [
85
+ "Q21.0",
86
+ "Q21.1",
87
+ "Q21.2",
88
+ "Q21.3",
89
+ "Q22.1",
90
+ "Q25.0",
91
+ "Q25.6",
92
+ "I37.0",
93
+ "I34.0",
94
+ "Q24.9"
95
+ ]
96
+ },
97
+ "architecture": {
98
+ "type": "Enhanced 1D CNN",
99
+ "blocks": "64\u2192128\u2192256\u2192512 filters",
100
+ "features": [
101
+ "Squeeze-Excitation",
102
+ "Temporal Attention",
103
+ "Residual Connections"
104
+ ],
105
+ "pooling": "Adaptive (Global Avg + Max)",
106
+ "total_params": 380627
107
+ },
108
+ "training": {
109
+ "epochs": 17,
110
+ "batch_size": 128,
111
+ "learning_rate": 0.0002,
112
+ "loss": "Focal Loss (alpha=0.25, gamma=2.0)",
113
+ "optimizer": "Adam"
114
+ },
115
+ "data": {
116
+ "total_samples": 14190,
117
+ "train_samples": 9933,
118
+ "val_samples": 2128,
119
+ "test_samples": 2129,
120
+ "sampling_rate": 500,
121
+ "num_channels": 12,
122
+ "variable_length": true,
123
+ "length_range": "5.0s - 30.0s"
124
+ },
125
+ "test_metrics": {
126
+ "subset_accuracy": 0.055894786284640674,
127
+ "hamming_loss": 0.07779782947269535,
128
+ "f1_micro": 0.08490840360569933,
129
+ "f1_macro": 0.012704413163746394,
130
+ "f1_weighted": 0.10957967386864802,
131
+ "precision_micro": 0.11986863711001643,
132
+ "recall_micro": 0.06573615488518686
133
+ },
134
+ "per_class_results": {
135
+ "Acute Myocarditis": {
136
+ "precision": 0.0,
137
+ "recall": 0.0,
138
+ "f1": 0.0,
139
+ "roc_auc": 0.545926688986179,
140
+ "support": 11
141
+ },
142
+ "Myocarditis Unspecified": {
143
+ "precision": 0.0,
144
+ "recall": 0.0,
145
+ "f1": 0.0,
146
+ "roc_auc": 0.5061110355612972,
147
+ "support": 28
148
+ },
149
+ "Dilated Cardiomyopathy": {
150
+ "precision": 0.0,
151
+ "recall": 0.0,
152
+ "f1": 0.0,
153
+ "roc_auc": 0.2941573792637623,
154
+ "support": 14
155
+ },
156
+ "Hypertrophic Cardiomyopathy": {
157
+ "precision": 0.0,
158
+ "recall": 0.0,
159
+ "f1": 0.0,
160
+ "roc_auc": 0.35010975227343993,
161
+ "support": 3
162
+ },
163
+ "Cardiomyopathy Unspecified": {
164
+ "precision": 0.0,
165
+ "recall": 0.0,
166
+ "f1": 0.0,
167
+ "roc_auc": 0.36835919134931827,
168
+ "support": 2
169
+ },
170
+ "Noncompaction Ventricular Myocardium": {
171
+ "precision": 0.0,
172
+ "recall": 0.0,
173
+ "f1": 0.0,
174
+ "roc_auc": 0.7858262778300408,
175
+ "support": 3
176
+ },
177
+ "Kawasaki Disease": {
178
+ "precision": 0.0,
179
+ "recall": 0.0,
180
+ "f1": 0.0,
181
+ "roc_auc": 0.4590394088669951,
182
+ "support": 29
183
+ },
184
+ "Ventricular Septal Defect": {
185
+ "precision": 0.0,
186
+ "recall": 0.0,
187
+ "f1": 0.0,
188
+ "roc_auc": 0.5319213433214829,
189
+ "support": 160
190
+ },
191
+ "Atrial Septal Defect": {
192
+ "precision": 0.0,
193
+ "recall": 0.0,
194
+ "f1": 0.0,
195
+ "roc_auc": 0.5427714279678025,
196
+ "support": 143
197
+ },
198
+ "Atrioventricular Septal Defect": {
199
+ "precision": 0.0,
200
+ "recall": 0.0,
201
+ "f1": 0.0,
202
+ "roc_auc": 0.5290356394129978,
203
+ "support": 9
204
+ },
205
+ "Tetralogy of Fallot": {
206
+ "precision": 0.010526315789473684,
207
+ "recall": 0.07142857142857142,
208
+ "f1": 0.01834862385321101,
209
+ "roc_auc": 0.3788247213779128,
210
+ "support": 14
211
+ },
212
+ "Pulmonary Valve Stenosis": {
213
+ "precision": 0.014705882352941176,
214
+ "recall": 0.2,
215
+ "f1": 0.0273972602739726,
216
+ "roc_auc": 0.6609227871939736,
217
+ "support": 5
218
+ },
219
+ "Patent Ductus Arteriosus": {
220
+ "precision": 0.0,
221
+ "recall": 0.0,
222
+ "f1": 0.0,
223
+ "roc_auc": 0.49539368912262505,
224
+ "support": 52
225
+ },
226
+ "Pulmonary Artery Stenosis": {
227
+ "precision": 0.0,
228
+ "recall": 0.0,
229
+ "f1": 0.0,
230
+ "roc_auc": 0.7461212976022567,
231
+ "support": 2
232
+ },
233
+ "Pulmonary Valve Regurgitation": {
234
+ "precision": 0.0,
235
+ "recall": 0.0,
236
+ "f1": 0.0,
237
+ "roc_auc": 0.39082809224318654,
238
+ "support": 9
239
+ },
240
+ "Mitral Valve Insufficiency": {
241
+ "precision": 0.0,
242
+ "recall": 0.0,
243
+ "f1": 0.0,
244
+ "roc_auc": 0.3574130361364404,
245
+ "support": 14
246
+ },
247
+ "Congenital Heart Malformation": {
248
+ "precision": 0.029411764705882353,
249
+ "recall": 0.22448979591836735,
250
+ "f1": 0.05200945626477541,
251
+ "roc_auc": 0.5164393642072214,
252
+ "support": 49
253
+ },
254
+ "Healthy": {
255
+ "precision": 0.7471910112359551,
256
+ "recall": 0.07945041816009558,
257
+ "f1": 0.14362850971922247,
258
+ "roc_auc": 0.48183137054104797,
259
+ "support": 1674
260
+ }
261
+ }
262
+ }