4rooms commited on
Commit
50c6b73
·
verified ·
1 Parent(s): 79b784a

Upload chambon2018/config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. chambon2018/config.json +32 -0
chambon2018/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_class": "physioex.models.chambon2018:Chambon2018Net",
3
+ "model_kwargs": {
4
+ "n_classes": 5,
5
+ "in_channels": 1,
6
+ "sf": 128,
7
+ "n_times": 3840,
8
+ "dropout": 0.25
9
+ },
10
+ "training": {
11
+ "dataset": "mass",
12
+ "dataset_kwargs": {
13
+ "cohort": 3
14
+ },
15
+ "channels": [
16
+ "EEG"
17
+ ],
18
+ "pipeline_preset": "raw",
19
+ "pipeline_kwargs": {
20
+ "target_fs": 128.0
21
+ },
22
+ "sequence_length": 3,
23
+ "max_epochs": 100,
24
+ "lr": 0.001,
25
+ "weight_decay": 0,
26
+ "batch_size": 128,
27
+ "loss": "CrossEntropyLoss",
28
+ "fold": 0,
29
+ "early_stopping_patience": 5
30
+ },
31
+ "reference": "Chambon et al. 2018 - A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series (arXiv:1707.03321)"
32
+ }