| { | |
| "model_class": "physioex.models.tinysleepnet:TinySleepNet", | |
| "model_kwargs": { | |
| "n_classes": 5, | |
| "in_chan": 1, | |
| "sf": 100, | |
| "n_rnn_units": 128, | |
| "n_rnn_layers": 1 | |
| }, | |
| "training": { | |
| "dataset": "sleepedf", | |
| "channels": [ | |
| "EEG" | |
| ], | |
| "pipeline_preset": "raw", | |
| "sequence_length": 20, | |
| "max_epochs": 200, | |
| "lr": 0.0001, | |
| "weight_decay": 0.001, | |
| "batch_size": 15, | |
| "loss": "CrossEntropyLoss", | |
| "fold": 0, | |
| "early_stopping_patience": 50 | |
| }, | |
| "reference": "Supratak & Guo 2020 - TinySleepNet: An Efficient Deep Learning Model for Sleep Stage Scoring based on Raw Single-Channel EEG" | |
| } |