previously trained model and results

archive/final_model.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3956b0731f0adf104e3f4bab2d1f533defb806b3bb4cbb1022fb1136a9332c11
+size 49516918

archive/model_results.json

@@ -0,0 +1,255 @@
+{
+  "model_name": "Enhanced_1D_CNN_19Classes_v3",
+  "version": "v03",
+  "created": "2025-11-30T19:57:52.553525",
+  "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": 4104363
+  },
+  "training": {
+    "epochs": 47,
+    "batch_size": 16,
+    "learning_rate": 0.001,
+    "loss": "Focal Loss (alpha=0.25, gamma=2.0)",
+    "optimizer": "Adam"
+  },
+  "data": {
+    "total_samples": 14189,
+    "train_samples": 9932,
+    "val_samples": 2128,
+    "test_samples": 2129,
+    "sampling_rate": 500,
+    "num_channels": 12,
+    "variable_length": true,
+    "length_range": "5.0s - 112.0s"
+  },
+  "test_metrics": {
+    "subset_accuracy": 0.6712071395021136,
+    "hamming_loss": 0.025240414328446762,
+    "f1_micro": 0.7367878319154422,
+    "f1_macro": 0.045103763907519925,
+    "f1_weighted": 0.648103905036729,
+    "precision_micro": 0.8676381299332119,
+    "recall_micro": 0.6402329749103942
+  },
+  "per_class_results": {
+    "Acute Myocarditis": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.5211892402076451,
+      "support": 10
+    },
+    "Myocarditis Unspecified": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.48553409940844494,
+      "support": 28
+    },
+    "Dilated Cardiomyopathy": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.6764427361853833,
+      "support": 17
+    },
+    "Hypertrophic Cardiomyopathy": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.8030089327691584,
+      "support": 2
+    },
+    "Cardiomyopathy Unspecified": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.7473700737949442,
+      "support": 6
+    },
+    "Kawasaki Disease": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.5302239288950252,
+      "support": 22
+    },
+    "Ventricular Septal Defect": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.7575439927588319,
+      "support": 152
+    },
+    "Atrial Septal Defect": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.772888934801216,
+      "support": 136
+    },
+    "Atrioventricular Septal Defect": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.874895178197065,
+      "support": 9
+    },
+    "Tetralogy of Fallot": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.9088430535798957,
+      "support": 20
+    },
+    "Pulmonary Valve Stenosis": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.7572941176470588,
+      "support": 4
+    },
+    "Patent Ductus Arteriosus": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.6519405777340901,
+      "support": 61
+    },
+    "Pulmonary Artery Stenosis": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.8650047036688617,
+      "support": 3
+    },
+    "Pulmonary Valve Regurgitation": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.775852752448497,
+      "support": 14
+    },
+    "Mitral Valve Insufficiency": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.7454115421002838,
+      "support": 15
+    },
+    "Congenital Heart Malformation": {
+      "precision": 0.0,
+      "recall": 0.0,
+      "f1": 0.0,
+      "roc_auc": 0.7265728300277244,
+      "support": 45
+    },
+    "Healthy": {
+      "precision": 0.8676381299332119,
+      "recall": 0.8465639810426541,
+      "f1": 0.8569715142428785,
+      "roc_auc": 0.7353521187305887,
+      "support": 1688
+    }
+  }
+}

archive/training_history.csv

@@ -0,0 +1,48 @@
+epoch,accuracy,auc,learning_rate,loss,precision,recall,val_accuracy,val_auc,val_loss,val_precision,val_recall
+0,0.8510822057723999,0.4787890911102295,0.0010000000474974513,1.72789466381073,0.17969286441802979,0.47941771149635315,0.9721260666847229,0.4329577088356018,0.26440921425819397,0.85161292552948,0.5953992009162903
+1,0.9643366932868958,0.474773108959198,0.0010000000474974513,0.24542951583862305,0.7770342230796814,0.49252867698669434,0.9539225697517395,0.49803853034973145,0.30018988251686096,0.9059633016586304,0.17816869914531708
+2,0.9659842252731323,0.49876469373703003,0.0010000000474974513,0.21423880755901337,0.802478551864624,0.5056396126747131,0.9451659321784973,0.5252382159233093,0.24668258428573608,0.0,0.0
+3,0.9661113619804382,0.49433064460754395,0.0010000000474974513,0.2034941166639328,0.7983798384666443,0.5130627751350403,0.9451659321784973,0.553851306438446,0.22968539595603943,0.0,0.0
+4,0.9670926332473755,0.5327081084251404,0.0010000000474974513,0.19187507033348083,0.8194934725761414,0.5147016048431396,0.975143313407898,0.5577508211135864,0.23844610154628754,0.7847744226455688,0.7532702088356018
+5,0.9670823216438293,0.5400435924530029,0.0010000000474974513,0.18725894391536713,0.818557620048523,0.5153764486312866,0.9748960733413696,0.43052807450294495,0.22700297832489014,0.7867366671562195,0.7437978982925415
+6,0.96604323387146,0.5071855783462524,0.0010000000474974513,0.1873648315668106,0.8088487386703491,0.5005302429199219,0.958102285861969,0.5455294251441956,0.18764373660087585,0.8944193124771118,0.2674785852432251
+7,0.9650725722312927,0.5008993744850159,0.0010000000474974513,0.18703939020633698,0.7958385348320007,0.490407794713974,0.9752423167228699,0.5057609677314758,0.19300174713134766,0.7892483472824097,0.748308539390564
+8,0.9647393226623535,0.49484121799468994,0.0010000000474974513,0.1862979680299759,0.7863853573799133,0.49223947525024414,0.975143313407898,0.5264214873313904,0.1875307410955429,0.7847744226455688,0.7532702088356018
+9,0.9651678204536438,0.4817586839199066,0.0010000000474974513,0.18455781042575836,0.7934816479682922,0.49522799253463745,0.9616640210151672,0.5660486817359924,0.17504636943340302,0.8668866753578186,0.3554352819919586
+10,0.965210497379303,0.49211809039115906,0.0010000000474974513,0.18226374685764313,0.8028953075408936,0.4865516126155853,0.9451906681060791,0.5198338627815247,0.18200837075710297,1.0,0.00045105998287908733
+11,0.9657667279243469,0.4943702220916748,0.0010000000474974513,0.1825098842382431,0.7873402833938599,0.5168225169181824,0.9569896459579468,0.5300279855728149,0.17633989453315735,0.878164529800415,0.25033828616142273
+12,0.9653910398483276,0.48058634996414185,0.0010000000474974513,0.18147894740104675,0.7882652878761292,0.5064108967781067,0.9550355076789856,0.5463641881942749,0.17462076246738434,0.8858800530433655,0.20658548176288605
+13,0.9659587740898132,0.5157257318496704,0.0010000000474974513,0.17977309226989746,0.7917836308479309,0.5165333151817322,0.975143313407898,0.590452253818512,0.17435185611248016,0.7847744226455688,0.7532702088356018
+14,0.9664880633354187,0.49744343757629395,0.0010000000474974513,0.1791478991508484,0.7961960434913635,0.5246312618255615,0.9496189951896667,0.540580153465271,0.17404460906982422,0.8629032373428345,0.09652683883905411
+15,0.9657347798347473,0.5000625252723694,0.0010000000474974513,0.1774548590183258,0.8007698059082031,0.5013978481292725,0.975143313407898,0.5797159075737,0.17855624854564667,0.7847744226455688,0.7532702088356018
+16,0.9667481184005737,0.5129356980323792,0.0010000000474974513,0.17581313848495483,0.8071963787078857,0.5190398097038269,0.975143313407898,0.5500307083129883,0.17516233026981354,0.7847744226455688,0.7532702088356018
+17,0.9668806195259094,0.509750247001648,0.0010000000474974513,0.17563903331756592,0.8052717447280884,0.5242456197738647,0.9724476933479309,0.5847578048706055,0.16985954344272614,0.8288610577583313,0.6269733905792236
+18,0.9673671722412109,0.5295020937919617,0.0010000000474974513,0.17329756915569305,0.8212162852287292,0.5194254517555237,0.9600563645362854,0.6110630035400391,0.16998961567878723,0.8858974575996399,0.31168246269226074
+19,0.9681046009063721,0.5378997921943665,0.0010000000474974513,0.17139823734760284,0.831556499004364,0.5263665318489075,0.9556294083595276,0.5834656357765198,0.17387500405311584,0.8824593424797058,0.22011727094650269
+20,0.9682894349098206,0.5591921210289001,0.0010000000474974513,0.1691010743379593,0.840074360370636,0.5226067900657654,0.9696033596992493,0.590927004814148,0.16552750766277313,0.8528571724891663,0.5385656356811523
+21,0.969577968120575,0.5847556591033936,0.0010000000474974513,0.1678459793329239,0.8435182571411133,0.5482502579689026,0.9738325476646423,0.6249508857727051,0.16573697328567505,0.8242865204811096,0.6644113659858704
+22,0.969026505947113,0.575284481048584,0.0010000000474974513,0.16915932297706604,0.8409638404846191,0.5383206605911255,0.9694055318832397,0.6571540832519531,0.1642550230026245,0.8740457892417908,0.5164636969566345
+23,0.9696144461631775,0.5994630455970764,0.0010000000474974513,0.16699130833148956,0.8447020649909973,0.5479610562324524,0.9735107421875,0.6563828587532043,0.16480490565299988,0.8390532732009888,0.6396030783653259
+24,0.9698423743247986,0.6023986339569092,0.0010000000474974513,0.16589638590812683,0.8500298857688904,0.5480574369430542,0.9745742678642273,0.6169835329055786,0.16769671440124512,0.8275482058525085,0.677492082118988
+25,0.9701665639877319,0.5937961339950562,0.0010000000474974513,0.16565196216106415,0.8528493046760559,0.5525884628295898,0.9722248911857605,0.646822452545166,0.16238129138946533,0.878808856010437,0.5723951458930969
+26,0.9706422686576843,0.6187255382537842,0.0010000000474974513,0.16316474974155426,0.8621309995651245,0.554612934589386,0.9753410220146179,0.6747590899467468,0.16741223633289337,0.83152174949646,0.6901217699050903
+27,0.9706855416297913,0.6245304942131042,0.0010000000474974513,0.1640329360961914,0.8579033017158508,0.5593367218971252,0.9663136601448059,0.6779999136924744,0.1652560830116272,0.9075309634208679,0.42940911650657654
+28,0.9703512191772461,0.6166345477104187,0.0010000000474974513,0.16351553797721863,0.8567801713943481,0.5530704855918884,0.9730904698371887,0.6744905710220337,0.16133542358875275,0.8625562191009521,0.6057735681533813
+29,0.9703306555747986,0.6215302348136902,0.0010000000474974513,0.16309238970279694,0.8587315678596497,0.5508531928062439,0.9750936627388,0.6896924376487732,0.16388675570487976,0.8353658318519592,0.6797474026679993
+30,0.9705994129180908,0.624535322189331,0.0010000000474974513,0.16237331926822662,0.8666970729827881,0.5496963262557983,0.9747721552848816,0.6806829571723938,0.1710323691368103,0.8341708779335022,0.6738836169242859
+31,0.9707751870155334,0.6282593011856079,0.0010000000474974513,0.1603950411081314,0.8715203404426575,0.5493106842041016,0.9743766188621521,0.7121614217758179,0.159336119890213,0.8651824593544006,0.6310329437255859
+32,0.9707913398742676,0.6573806405067444,0.0010000000474974513,0.16036613285541534,0.872833251953125,0.5485394597053528,0.9756624698638916,0.695950984954834,0.16716766357421875,0.817619800567627,0.7158322334289551
+33,0.9712041616439819,0.658597469329834,0.0010000000474974513,0.15952520072460175,0.878119707107544,0.55287766456604,0.9745497107505798,0.6628549695014954,0.16508089005947113,0.8253012299537659,0.6797474026679993
+34,0.9704206585884094,0.6633495092391968,0.0010000000474974513,0.16060425341129303,0.8744723796844482,0.5392847061157227,0.9750444293022156,0.6673092842102051,0.16927646100521088,0.8004975318908691,0.725755512714386
+35,0.9708438515663147,0.6568770408630371,0.0010000000474974513,0.15939481556415558,0.8794204592704773,0.5442013144493103,0.9746731519699097,0.6645341515541077,0.16298921406269073,0.8379603624343872,0.667117714881897
+36,0.9707704782485962,0.6505053043365479,0.0010000000474974513,0.15937505662441254,0.8784478902816772,0.5434300303459167,0.9754646420478821,0.6705322861671448,0.16291269659996033,0.8316188454627991,0.6928281188011169
+37,0.971077024936676,0.6552930474281311,0.0005000000237487257,0.15848912298679352,0.8791853189468384,0.5493106842041016,0.9754895567893982,0.6664952635765076,0.16160069406032562,0.8398004174232483,0.6833558678627014
+38,0.971156895160675,0.6667534708976746,0.0005000000237487257,0.1574985533952713,0.8806362152099609,0.5497927069664001,0.9747227430343628,0.6824701428413391,0.16209770739078522,0.8396816253662109,0.6662155985832214
+39,0.9712361693382263,0.6576641201972961,0.0005000000237487257,0.15879565477371216,0.876848042011261,0.554612934589386,0.9722498655319214,0.7065017819404602,0.1571519374847412,0.8831350803375244,0.5692377090454102
+40,0.9703616499900818,0.6631851196289062,0.0005000000237487257,0.15829308331012726,0.8719265460968018,0.5401523113250732,0.975687563419342,0.701470673084259,0.1607586145401001,0.8466292023658752,0.6797474026679993
+41,0.9707229137420654,0.661217212677002,0.0005000000237487257,0.1584259271621704,0.8776877522468567,0.5430444478988647,0.9762069582939148,0.701678991317749,0.16394875943660736,0.809876561164856,0.739738404750824
+42,0.9707803130149841,0.6525394916534424,0.0005000000237487257,0.15819983184337616,0.8787217736244202,0.5434300303459167,0.9750937819480896,0.6832001209259033,0.16189807653427124,0.857988178730011,0.6540369987487793
+43,0.9705634117126465,0.6525664329528809,0.0005000000237487257,0.15789981186389923,0.8829888701438904,0.535428524017334,0.9756627082824707,0.6969619989395142,0.16236679255962372,0.8411732316017151,0.6856111884117126
+44,0.9704945087432861,0.6497383713722229,0.0005000000237487257,0.15794014930725098,0.8833572864532471,0.533693253993988,0.975143313407898,0.6731274127960205,0.16921544075012207,0.7850423455238342,0.7528191208839417
+45,0.9711570739746094,0.6567102670669556,0.0002500000118743628,0.15756560862064362,0.8822891116142273,0.548443078994751,0.975143313407898,0.7011749148368835,0.16688035428524017,0.7853107452392578,0.7523680925369263
+46,0.9710562229156494,0.6691934466362,0.0002500000118743628,0.1571454405784607,0.8813480138778687,0.547093391418457,0.9751185178756714,0.6770845651626587,0.1702290028333664,0.7849411964416504,0.7523680925369263