| { | |
| "model_class": "physioex.models.tsinalis:TsinalisCNN", | |
| "model_kwargs": { | |
| "n_classes": 5, | |
| "sfreq": 100, | |
| "n_filters_c1": 20, | |
| "n_filters_c2": 400, | |
| "fc_size": 500, | |
| "dropout": 0.5 | |
| }, | |
| "training": { | |
| "dataset": "sleepedf", | |
| "channels": [ | |
| "EEG" | |
| ], | |
| "pipeline_preset": "identity", | |
| "sequence_length": 5, | |
| "max_epochs": 100, | |
| "lr": 0.001, | |
| "weight_decay": 0.0001, | |
| "batch_size": 32, | |
| "loss": "CrossEntropyLoss", | |
| "fold": 0, | |
| "early_stopping_patience": 20, | |
| "optimizer": "SGD" | |
| }, | |
| "reference": "Tsinalis et al. 2016 - Automatic Sleep Stage Scoring with Single-Channel EEG Using Convolutional Neural Networks (arXiv:1610.01683)" | |
| } |