-
Models.py in this project contains three neural network: ** CNN + LSTM ** Self-Attention + LSTM ** Mix Self-Attention and CNN + LSTM
-
To run the project:
python Mix.py -i [number of channels]
Some directories and path are expected to be created and changed
- The project is developed with dataset "Sleep EDF Database Expanded": https://physionet.org/content/sleep-edfx/1.0.0/
Originally developed on: https://github.com/GuanLab/Sleep-Stage-Auto-annotation We adjust the project to apply it to our clinic data (not available)