garethjns / Kaggle-EEG

Seizure prediction from EEG data using machine learning. 3rd place solution for Kaggle/Uni Melbourne seizure prediction competition.
101 stars 29 forks source link

Feature information not saved in seizureModels #3

Open garethjns opened 7 years ago

garethjns commented 7 years ago

Some useful information about features is not saved in seizureModel object, and need to be reset manually in predict script.

use - Structure that lists features included in model. params.divS - Field containing list of epoch window lengths used to generate features.

Either save these in the seizureModel object during training, or save entire features object in seizureModel (minus actual data) for reference.