/sleep-stage-classification-for-image

Standardized Image-Based Polysomnography Database and Deep Learning Algorithm for Sleep Stage Classification

Primary LanguagePythonApache License 2.0Apache-2.0

Sleep Stage Classification

Standardized Image-Based Polysomnography Database and Deep Learning Algorithm for Sleep Stage Classification

1

Dataset

Our Dataset

Train

python 1_train_conv.py
python 2_train_lstm.py

Train with FOLD

python 1_train_conv_fold.py
python 2_train_lstm_fold.py

Test

python 3_test.py