Hugging Face's logo

4rooms
/
physioex

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