| { | |
| "model_class": "physioex.models.sleeptransformer:SleepTransformer", | |
| "model_kwargs": { | |
| "n_classes": 5, | |
| "in_chan": 1, | |
| "d_model": 128, | |
| "n_heads": 8, | |
| "n_epoch_layers": 4, | |
| "n_seq_layers": 4, | |
| "d_ff": 1024, | |
| "d_clf": 1024, | |
| "dropout": 0.1, | |
| "attention_size": 128 | |
| }, | |
| "training": { | |
| "dataset": "shhs", | |
| "dataset_kwargs": { | |
| "visit": 1 | |
| }, | |
| "channels": [ | |
| "EEG" | |
| ], | |
| "pipeline_preset": "seqsleepnet", | |
| "sequence_length": 21, | |
| "max_epochs": 50, | |
| "lr": 0.0001, | |
| "weight_decay": 0, | |
| "batch_size": 32, | |
| "loss": "CrossEntropyLoss", | |
| "fold": 0, | |
| "early_stopping_patience": 10 | |
| }, | |
| "reference": "Phan et al. 2022 - SleepTransformer: Automatic Sleep Staging with Interpretability and Uncertainty Quantification (arXiv:2105.11043)" | |
| } |