MIT-BIH Arrhythmia database

• Reproduce the results

A pre-trained model is provided in the model directory. To reproduce our results, you should, first, download the MIT-BIH arrhythmia database from the above link and save it in the dataset directory. Then, execute preprocessing.py to obtain the training and test dataset. After that, just run pre-training.py and you will get the results of our paper.

• Re-train the model

Just replace the last step of run pre-training.py with main.py.

Our CNN is implemented using PyTorch v1.4.0 and trained on the NVIDIA GeForce RTX 2080Ti graphical processing unit. Noted that different versions of PyTorch and CUDA may cause the results to be slightly different from the results in the article.

The paper is submitted to the journal for review.

If you have any questions, please email to: wtustc@mail.ustc.edu.cn