Skip to content

Python framework for designing and training Deep Learning models to reconstruct 12-lead Electrocardiograms from 3-lead inputs and classify clinical conditions effectively.

Notifications You must be signed in to change notification settings

jingweili87/DeepECG-Reconstructor

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DeepECG-Reconstructor

This Python framework provides tools for designing and training deep-learning models capable of reconstructing 12-lead Electrocardiograms (ECGs) from 3-lead inputs and assessing clinical conditions accurately.

Main Functions

explore_dataset.py

  • Perform explorative analysis on the dataset, associating elements with multiple clinical labels and computing statistical metrics for different classes.

process_dataset.py

  • Discard corrupted elements and divide remaining ones into training, validation, and testing sets.

single_reconstruction.py

  • Design and train a deep-learning model that reconstructs 12-lead ECGs using mathematical differences between reconstructed signals and originals as loss functions.

single_classification.py

  • Train a model to classify ECGs into specific clinical classes using detection accuracy as the optimization metric.

single_recon_classif.py

  • Train a model to simultaneously reconstruct ECGs and classify signals, optimizing both reconstruction loss and classification probability.

multi_reconstruction.py

  • Train multiple models to perform ECG reconstruction using mathematical differences as loss functions.

multi_classification.py

  • Train multiple models to classify ECG signals, optimizing detection accuracy.

multi_recon_classif.py

  • Train multiple models to reconstruct ECG signals while optimizing class association probabilities.

Additional Functions

generate_dataclass.py

  • Create new data classes as unions or intersections of existing clinical labels.

clean_dataclass.py

  • Identify corrupted vs. cleaned elements within a class and gather insights into their differences.

analyze_dataclass.py

  • Compute probabilities of association with different clinical labels within a class.

process_dataclass.py

  • Execute cleaning and analysis sequentially over a specific class of data elements.

About

Python framework for designing and training Deep Learning models to reconstruct 12-lead Electrocardiograms from 3-lead inputs and classify clinical conditions effectively.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages