Hugging Face's logo

4rooms
/
physioex

PyTorch
English
sleep-staging
eeg
physioex
Model card Files
xet
 Community
physioex
File size: 815 Bytes
9685179
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "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)"
}