{
  "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"
}