ECG-Disease-Classifier / model_results.json
kiril-buga's picture
Add model metadata
e72c065 verified
{
"model_name": "1D_CNN_19Classes_multilabel_19v3",
"version": "v03_balanced",
"created": "2025-12-07T14:07:16.688070",
"num_classes": 19,
"disease_names": [
"Fulminant/Viral Myocarditis",
"Acute Myocarditis",
"Myocarditis Unspecified",
"Dilated Cardiomyopathy",
"Hypertrophic Cardiomyopathy",
"Cardiomyopathy Unspecified",
"Noncompaction Ventricular Myocardium",
"Kawasaki Disease",
"Ventricular Septal Defect",
"Atrial Septal Defect",
"Atrioventricular Septal Defect",
"Tetralogy of Fallot",
"Pulmonary Valve Stenosis",
"Patent Ductus Arteriosus",
"Pulmonary Artery Stenosis",
"Pulmonary Valve Regurgitation",
"Mitral Valve Insufficiency",
"Congenital Heart Malformation",
"Healthy"
],
"icd_codes": [
"I40.0",
"I40.9",
"I51.4",
"I42.0",
"I42.2",
"I42.9",
"Q24.8",
"M30.3",
"Q21.0",
"Q21.1",
"Q21.2",
"Q21.3",
"Q22.1",
"Q25.0",
"Q25.6",
"I37.0",
"I34.0",
"Q24.9",
"Healthy"
],
"icd_to_disease_name": {
"I40.0": "Fulminant/Viral Myocarditis",
"I40.9": "Acute Myocarditis",
"I51.4": "Myocarditis Unspecified",
"I42.0": "Dilated Cardiomyopathy",
"I42.2": "Hypertrophic Cardiomyopathy",
"I42.9": "Cardiomyopathy Unspecified",
"Q24.8": "Noncompaction Ventricular Myocardium",
"M30.3": "Kawasaki Disease",
"Q21.0": "Ventricular Septal Defect",
"Q21.1": "Atrial Septal Defect",
"Q21.2": "Atrioventricular Septal Defect",
"Q21.3": "Tetralogy of Fallot",
"Q22.1": "Pulmonary Valve Stenosis",
"Q25.0": "Patent Ductus Arteriosus",
"Q25.6": "Pulmonary Artery Stenosis",
"I37.0": "Pulmonary Valve Regurgitation",
"I34.0": "Mitral Valve Insufficiency",
"Q24.9": "Congenital Heart Malformation",
"Healthy": "Healthy"
},
"disease_groups": {
"Myocarditis": [
"I40.0",
"I40.9",
"I51.4"
],
"Cardiomyopathy": [
"I42.0",
"I42.2",
"I42.9",
"Q24.8"
],
"Kawasaki": [
"M30.3"
],
"CHD": [
"Q21.0",
"Q21.1",
"Q21.2",
"Q21.3",
"Q22.1",
"Q25.0",
"Q25.6",
"I37.0",
"I34.0",
"Q24.9"
]
},
"architecture": {
"type": "Enhanced 1D CNN",
"blocks": "64\u2192128\u2192256\u2192512 filters",
"features": [
"Squeeze-Excitation",
"Temporal Attention",
"Residual Connections"
],
"pooling": "Adaptive (Global Avg + Max)",
"total_params": 380627
},
"training": {
"epochs": 17,
"batch_size": 128,
"learning_rate": 0.0002,
"loss": "Focal Loss (alpha=0.25, gamma=2.0)",
"optimizer": "Adam"
},
"data": {
"total_samples": 14190,
"train_samples": 9933,
"val_samples": 2128,
"test_samples": 2129,
"sampling_rate": 500,
"num_channels": 12,
"variable_length": true,
"length_range": "5.0s - 30.0s"
},
"test_metrics": {
"subset_accuracy": 0.055894786284640674,
"hamming_loss": 0.07779782947269535,
"f1_micro": 0.08490840360569933,
"f1_macro": 0.012704413163746394,
"f1_weighted": 0.10957967386864802,
"precision_micro": 0.11986863711001643,
"recall_micro": 0.06573615488518686
},
"per_class_results": {
"Acute Myocarditis": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.545926688986179,
"support": 11
},
"Myocarditis Unspecified": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.5061110355612972,
"support": 28
},
"Dilated Cardiomyopathy": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.2941573792637623,
"support": 14
},
"Hypertrophic Cardiomyopathy": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.35010975227343993,
"support": 3
},
"Cardiomyopathy Unspecified": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.36835919134931827,
"support": 2
},
"Noncompaction Ventricular Myocardium": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.7858262778300408,
"support": 3
},
"Kawasaki Disease": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.4590394088669951,
"support": 29
},
"Ventricular Septal Defect": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.5319213433214829,
"support": 160
},
"Atrial Septal Defect": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.5427714279678025,
"support": 143
},
"Atrioventricular Septal Defect": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.5290356394129978,
"support": 9
},
"Tetralogy of Fallot": {
"precision": 0.010526315789473684,
"recall": 0.07142857142857142,
"f1": 0.01834862385321101,
"roc_auc": 0.3788247213779128,
"support": 14
},
"Pulmonary Valve Stenosis": {
"precision": 0.014705882352941176,
"recall": 0.2,
"f1": 0.0273972602739726,
"roc_auc": 0.6609227871939736,
"support": 5
},
"Patent Ductus Arteriosus": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.49539368912262505,
"support": 52
},
"Pulmonary Artery Stenosis": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.7461212976022567,
"support": 2
},
"Pulmonary Valve Regurgitation": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.39082809224318654,
"support": 9
},
"Mitral Valve Insufficiency": {
"precision": 0.0,
"recall": 0.0,
"f1": 0.0,
"roc_auc": 0.3574130361364404,
"support": 14
},
"Congenital Heart Malformation": {
"precision": 0.029411764705882353,
"recall": 0.22448979591836735,
"f1": 0.05200945626477541,
"roc_auc": 0.5164393642072214,
"support": 49
},
"Healthy": {
"precision": 0.7471910112359551,
"recall": 0.07945041816009558,
"f1": 0.14362850971922247,
"roc_auc": 0.48183137054104797,
"support": 1674
}
}
}