File size: 870 Bytes
ed875c5 35b77c1 ed875c5 35b77c1 ed875c5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | {
"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"
} |