/AR-detection-with-ECG

The repository of the source code of "Correlating Electrocardiograms with Echocardiographic Parameters in Hemodynamically-Significant Aortic Regurgitation Using Deep Learning"

Primary LanguagePythonMIT LicenseMIT

About

The source code of "Correlating Electrocardiograms with Echocardiographic Parameters in Hemodynamically-Significant Aortic Regurgitation Using Deep Learning"

3-Central illustration

Files

  • Model training and validation: main.py
  • Dataset preperation: dataset.py
  • Testing code: test.py
  • Shapley value calculation and visualization: SHAP.py