search

akaraspt / deepsleepnet

DeepSleepNet: a Model for Automatic Sleep Stage Scoring based on Raw Single-Channel EEG
Apache License 2.0
398 stars 153 forks source link

Is there a pre-trained model in the files? #10

Closed JunshuaiFeng closed 6 years ago

JunshuaiFeng commented 6 years ago

I am currently referring this project for one of my ML classes and I heard it takes about one week to train the model, and I am still working on getting the MASS dataset. Was the trained model generated and saved somewhere in the files or I have to train it every time? It would be awesome if there is a pre-trained model, then I can maybe skip the training part to save some time for the predicting part.

Thanks in advance!

akaraspt commented 6 years ago

Thank you for your suggestions.

We have a plan to upload the pre-trained models, so that you can evaluate our model without training the model for all CV folds.

We will let you know once they have been uploaded.

JunshuaiFeng commented 6 years ago

Thank you very much!

wusuowei11 commented 2 years ago

您好，现在预训练模型有上传么