1D vision transformer (vit) for ecg interpretable classification
Code for my blog post
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