python main.py
You can edit the configuration file 'config.py' to change train options.
Default data location: '../data'
PTB-XL: PTB-XL, a large publicly available electrocardiography dataset v1.0.1 (physionet.org).
CPSC: The China Physiological Signal Challenge 2018 (icbeb.org).
- matplotlib>=3.5.2
- numpy>=1.22.3
- pandas>=1.4.2
- scikit_learn>=1.1.0
- torch>=1.11.0+cu113
- tqdm>=4.64.0
- wandb>=0.12.16
- wfdb>=3.4.1