Hugging Face's logo

4rooms
/
physioex

PyTorch
English
sleep-staging
eeg
physioex
Model card Files
xet
 Community
physioex
File size: 771 Bytes
50c6b73
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "model_class": "physioex.models.chambon2018:Chambon2018Net",
  "model_kwargs": {
    "n_classes": 5,
    "in_channels": 1,
    "sf": 128,
    "n_times": 3840,
    "dropout": 0.25
  },
  "training": {
    "dataset": "mass",
    "dataset_kwargs": {
      "cohort": 3
    },
    "channels": [
      "EEG"
    ],
    "pipeline_preset": "raw",
    "pipeline_kwargs": {
      "target_fs": 128.0
    },
    "sequence_length": 3,
    "max_epochs": 100,
    "lr": 0.001,
    "weight_decay": 0,
    "batch_size": 128,
    "loss": "CrossEntropyLoss",
    "fold": 0,
    "early_stopping_patience": 5
  },
  "reference": "Chambon et al. 2018 - A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series (arXiv:1707.03321)"
}