DS3001 Course Project

Author: Jiacheng Wang, Weiheng Li

  • Our project is to make predicting classification based on one-channel ECG signal using BiLSTM and CNN

  • Data folder conatins ECG data from MIT-BIH ECG database

  • Initail_Resources contains the papers inspired us.

  • src folder is our code to implement different architecture of our project.

    • data.py is to generate the dataloader for pytorch
    • main.py is the main entrance to run different algorithm
    • train.py is to define the network achitecture
    • utils.py contains pre-processing steps
  • Instructor: Cristina Savin