How cvxtz/ecg_heartbeat_classification works
This repository is a set of standalone ECG heartbeat classification experiments, not a deployed service. The main workflow loads preprocessed MIT-BIH or PTB heartbeat CSVs, trains a 1D convolutional classifier, and reports test accuracy and F1; PTB transfer variants initialize from MIT-BIH weights, with one freezing convolutional layers. A separate EDA script summarizes and plots test heartbeats. Each experiment implements its own data preparation, model, and evaluation inline; the graph keeps those workflows together. Evidence is limited to the README and the sampled scripts.
Main entry points: baseline_ptbdb_transfer_freeze.py Frozen transfer · baseline_ptbdb_transfer_fullupdate.py Full-update transfer · baseline_mitbih.py MIT-BIH classifier · baseline_ptbdb.py PTB classifier
Components 10
Classification experiments
- MIT-BIH classifier
code/baseline_mitbih.py - PTB classifier
code/baseline_ptbdb.py - Frozen transfer
code/baseline_ptbdb_transfer_freeze.py - Full-update transfer
code/baseline_ptbdb_transfer_fullupdate.py
Dataset analysis
- Heartbeat EDA
code/eda.py
Data and weights
- MIT-BIH CSVs
- PTB CSVs
- MIT-BIH weights
- PTB checkpoints
Other components
- Researcher
Connections 16
- Researcher → MIT-BIH classifier: runs experiment
README.md - Researcher → PTB classifier: runs experiment
README.md - Researcher → Frozen transfer: runs experiment
README.md - Researcher → Full-update transfer: runs experiment
README.md - Researcher → Heartbeat EDA: runs analysis
code/eda.py - MIT-BIH classifier → MIT-BIH CSVs: reads CSVs
code/baseline_mitbih.py - MIT-BIH classifier → MIT-BIH weights: writes checkpoint
code/baseline_mitbih.py - PTB classifier → PTB CSVs: reads CSVs
code/baseline_ptbdb.py - PTB classifier → PTB checkpoints: writes checkpoint
code/baseline_ptbdb.py - Frozen transfer → PTB CSVs: reads CSVs
code/baseline_ptbdb_transfer_freeze.py - Frozen transfer → MIT-BIH weights: loads weights
code/baseline_ptbdb_transfer_freeze.py - Frozen transfer → PTB checkpoints: writes checkpoint
code/baseline_ptbdb_transfer_freeze.py - Full-update transfer → PTB CSVs: reads CSVs
code/baseline_ptbdb_transfer_fullupdate.py - Full-update transfer → MIT-BIH weights: loads weights
code/baseline_ptbdb_transfer_fullupdate.py - Full-update transfer → PTB checkpoints: writes checkpoint
code/baseline_ptbdb_transfer_fullupdate.py - Heartbeat EDA → MIT-BIH CSVs: reads test CSV
code/eda.py