ecg_vit

1D vision transformer (vit) for ecg interpretable classification
Code for my blog post

Installation

python version 3.10.5
dependencies in requirements.txt

python train.py /<path>/a-large-scale-12-lead-electrocardiogram-database-for-arrhythmia-study-1.0.0/

python plot_attention.py  <wfdb_file> <model_weights_path> <lead_index>`

For example:

python plot_attention.py a-large-scale-12-lead-electrocardiogram-database-for-arrhythmia-study-1.0.0/WFDBRecords/06/066/JS05865 vit_best/ 6

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
README.md		README.md
data.py		data.py
plot.py		plot.py
plot_attention.py		plot_attention.py
requirements.txt		requirements.txt
train.py		train.py
vit.py		vit.py