ECG Classifier - Class Project for Statistical Learning 2020
Data
You can obtain datasets from StatLearning SJTU 2020.
Environment
- Python 3.5+
- tensorflow
- sklearn
- pywt
- seaborn
- biosppy
- scipy
Run it
- Place the
ecg.py
in the dataset directory - Run the python code with one parameter
# using CNN
$ python ecg.py cnn
# using SVM
$ python ecg.py svm
# using kNN
$ python ecg.py knn
- You will get a trained model and the classification result in
.csv
format.