| { | |
| "model_class": "physioex.models.seqsleepnet:SeqSleepNet", | |
| "model_kwargs": { | |
| "n_classes": 5, | |
| "in_chan": 1, | |
| "F": 129, | |
| "D": 32, | |
| "nfft": 256, | |
| "lowfreq": 0, | |
| "highfreq": 50, | |
| "fs": 100, | |
| "seqnhidden1": 64, | |
| "seqnlayer1": 4, | |
| "attentionsize": 32, | |
| "seqnhidden2": 64, | |
| "seqnlayer2": 4 | |
| }, | |
| "training": { | |
| "dataset": "mass", | |
| "dataset_cohorts": [ | |
| 1, | |
| 2, | |
| 3, | |
| 4, | |
| 5 | |
| ], | |
| "channels": [ | |
| "EEG" | |
| ], | |
| "pipeline_preset": "seqsleepnet", | |
| "sequence_length": 20, | |
| "max_epochs": 10, | |
| "lr": 0.0001, | |
| "weight_decay": 0, | |
| "batch_size": 32, | |
| "loss": "CrossEntropyLoss", | |
| "fold": 0, | |
| "early_stopping_patience": 10 | |
| }, | |
| "reference": "Phan et al. 2019 - SeqSleepNet: End-to-End Hierarchical Recurrent Neural Network for Sequence-to-Sequence Automatic Sleep Staging" | |
| } |