search

skjerns / AutoSleepScorer

An open-source sleep stage classification Python package
GNU Affero General Public License v3.0
103 stars 22 forks source link

How to train the model #1

Open shqing opened 6 years ago

shqing commented 6 years ago

Hi, if I download the dataset, how can I to train the model ?

skjerns commented 6 years ago

Code to train the model is stored at https://github.com/skjerns/AutoSleepScorerDev

Emphasis on stored as I had no time yet to clean up the repository to make is usable, but it's on my list. I'll let you know if that changes!

Until then you can find a pre-trained model integrated.

skjerns commented 6 years ago

@shqing I created a small script run_sample.py that should help you to get started:

https://github.com/skjerns/AutoSleepScorerDev/blob/master/run_sample.py

just clone the https://github.com/skjerns/AutoSleepScorerDev/ repo. Hope it works.

edit: just checked it: I think I changed some settings so that results deviate from the final results. I'll look into it later.