4rooms commited on
Commit
d6f6eb3
·
verified ·
1 Parent(s): 9570dab

Upload coresleep-kontras/config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. coresleep-kontras/config.json +32 -0
coresleep-kontras/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_class": "physioex.models.coresleep:CoReSleep",
3
+ "model_kwargs": {
4
+ "n_classes": 5,
5
+ "in_chan": 2,
6
+ "F": 129,
7
+ "d_model": 128,
8
+ "n_heads": 8,
9
+ "n_inner_layers": 4,
10
+ "n_outer_layers": 4,
11
+ "d_ff": 1024,
12
+ "dropout": 0.3
13
+ },
14
+ "training": {
15
+ "dataset": "shhs",
16
+ "dataset_kwargs": {
17
+ "visit": 1
18
+ },
19
+ "channels": [
20
+ "EEG",
21
+ "EOG"
22
+ ],
23
+ "pipeline_preset": "coresleep",
24
+ "sequence_length": 21,
25
+ "max_epochs": 50,
26
+ "lr": 0.0001,
27
+ "weight_decay": 0.0001,
28
+ "batch_size": 16,
29
+ "fold": 0
30
+ },
31
+ "reference": "Kontras et al. 2024 - CoRe-Sleep: A Multimodal Fusion Framework"
32
+ }