Data comes from the MIT-BIH database and is prepared by this project: GitHub.
There are three models with a similar number of trainable parameters:
- 1-block ResNet
- Neural ODE
- Deep Equilibrium Network
Simply run the Jupyter Notebook ecg.ipynb
. Model training will take a while for the Neural ODE and DEQ models.
Requirements
numpy
pandas
scikit-learn
pytorch
torchinfo
torchdiffeq
python_experiments
│ ecg.ipynb - Notebook for training and testing models
│ models.py - Model functions
│ utils.py - Other helper functions
└───data
mitdb_360_test.csv - Training data
mitdb_360_train.csv - Test data