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
439b592
·
verified ·
1 Parent(s): 6703645

Add model metadata

Browse files
Files changed (1) hide show
  1. model_results.json +255 -0
model_results.json ADDED
@@ -0,0 +1,255 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Enhanced_1D_CNN_19Classes_v3",
3
+ "version": "v03",
4
+ "created": "2025-11-30T19:57:52.553525",
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": 4104363
107
+ },
108
+ "training": {
109
+ "epochs": 47,
110
+ "batch_size": 16,
111
+ "learning_rate": 0.001,
112
+ "loss": "Focal Loss (alpha=0.25, gamma=2.0)",
113
+ "optimizer": "Adam"
114
+ },
115
+ "data": {
116
+ "total_samples": 14189,
117
+ "train_samples": 9932,
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 - 112.0s"
124
+ },
125
+ "test_metrics": {
126
+ "subset_accuracy": 0.6712071395021136,
127
+ "hamming_loss": 0.025240414328446762,
128
+ "f1_micro": 0.7367878319154422,
129
+ "f1_macro": 0.045103763907519925,
130
+ "f1_weighted": 0.648103905036729,
131
+ "precision_micro": 0.8676381299332119,
132
+ "recall_micro": 0.6402329749103942
133
+ },
134
+ "per_class_results": {
135
+ "Acute Myocarditis": {
136
+ "precision": 0.0,
137
+ "recall": 0.0,
138
+ "f1": 0.0,
139
+ "roc_auc": 0.5211892402076451,
140
+ "support": 10
141
+ },
142
+ "Myocarditis Unspecified": {
143
+ "precision": 0.0,
144
+ "recall": 0.0,
145
+ "f1": 0.0,
146
+ "roc_auc": 0.48553409940844494,
147
+ "support": 28
148
+ },
149
+ "Dilated Cardiomyopathy": {
150
+ "precision": 0.0,
151
+ "recall": 0.0,
152
+ "f1": 0.0,
153
+ "roc_auc": 0.6764427361853833,
154
+ "support": 17
155
+ },
156
+ "Hypertrophic Cardiomyopathy": {
157
+ "precision": 0.0,
158
+ "recall": 0.0,
159
+ "f1": 0.0,
160
+ "roc_auc": 0.8030089327691584,
161
+ "support": 2
162
+ },
163
+ "Cardiomyopathy Unspecified": {
164
+ "precision": 0.0,
165
+ "recall": 0.0,
166
+ "f1": 0.0,
167
+ "roc_auc": 0.7473700737949442,
168
+ "support": 6
169
+ },
170
+ "Kawasaki Disease": {
171
+ "precision": 0.0,
172
+ "recall": 0.0,
173
+ "f1": 0.0,
174
+ "roc_auc": 0.5302239288950252,
175
+ "support": 22
176
+ },
177
+ "Ventricular Septal Defect": {
178
+ "precision": 0.0,
179
+ "recall": 0.0,
180
+ "f1": 0.0,
181
+ "roc_auc": 0.7575439927588319,
182
+ "support": 152
183
+ },
184
+ "Atrial Septal Defect": {
185
+ "precision": 0.0,
186
+ "recall": 0.0,
187
+ "f1": 0.0,
188
+ "roc_auc": 0.772888934801216,
189
+ "support": 136
190
+ },
191
+ "Atrioventricular Septal Defect": {
192
+ "precision": 0.0,
193
+ "recall": 0.0,
194
+ "f1": 0.0,
195
+ "roc_auc": 0.874895178197065,
196
+ "support": 9
197
+ },
198
+ "Tetralogy of Fallot": {
199
+ "precision": 0.0,
200
+ "recall": 0.0,
201
+ "f1": 0.0,
202
+ "roc_auc": 0.9088430535798957,
203
+ "support": 20
204
+ },
205
+ "Pulmonary Valve Stenosis": {
206
+ "precision": 0.0,
207
+ "recall": 0.0,
208
+ "f1": 0.0,
209
+ "roc_auc": 0.7572941176470588,
210
+ "support": 4
211
+ },
212
+ "Patent Ductus Arteriosus": {
213
+ "precision": 0.0,
214
+ "recall": 0.0,
215
+ "f1": 0.0,
216
+ "roc_auc": 0.6519405777340901,
217
+ "support": 61
218
+ },
219
+ "Pulmonary Artery Stenosis": {
220
+ "precision": 0.0,
221
+ "recall": 0.0,
222
+ "f1": 0.0,
223
+ "roc_auc": 0.8650047036688617,
224
+ "support": 3
225
+ },
226
+ "Pulmonary Valve Regurgitation": {
227
+ "precision": 0.0,
228
+ "recall": 0.0,
229
+ "f1": 0.0,
230
+ "roc_auc": 0.775852752448497,
231
+ "support": 14
232
+ },
233
+ "Mitral Valve Insufficiency": {
234
+ "precision": 0.0,
235
+ "recall": 0.0,
236
+ "f1": 0.0,
237
+ "roc_auc": 0.7454115421002838,
238
+ "support": 15
239
+ },
240
+ "Congenital Heart Malformation": {
241
+ "precision": 0.0,
242
+ "recall": 0.0,
243
+ "f1": 0.0,
244
+ "roc_auc": 0.7265728300277244,
245
+ "support": 45
246
+ },
247
+ "Healthy": {
248
+ "precision": 0.8676381299332119,
249
+ "recall": 0.8465639810426541,
250
+ "f1": 0.8569715142428785,
251
+ "roc_auc": 0.7353521187305887,
252
+ "support": 1688
253
+ }
254
+ }
255
+ }