4rooms commited on
Commit
98506be
·
verified ·
1 Parent(s): c337cc8

Upload lseqsleepnet-phan/config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. lseqsleepnet-phan/config.json +38 -0
lseqsleepnet-phan/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_class": "physioex.models.lseqsleepnet:LSeqSleepNet",
3
+ "model_kwargs": {
4
+ "n_classes": 5,
5
+ "in_chan": 1,
6
+ "F": 129,
7
+ "D": 32,
8
+ "nfft": 256,
9
+ "sf": 100,
10
+ "lowfreq": 0,
11
+ "highfreq": 50,
12
+ "epoch_hidden": 64,
13
+ "epoch_attention": 64,
14
+ "B": 10,
15
+ "K": 20,
16
+ "seq_hidden_ss": 64,
17
+ "seq_hidden_ms": 64,
18
+ "d_clf": 512,
19
+ "dropout": 0.1
20
+ },
21
+ "training": {
22
+ "dataset": "shhs",
23
+ "dataset_kwargs": {
24
+ "visit": 1
25
+ },
26
+ "channels": [
27
+ "EEG"
28
+ ],
29
+ "pipeline_preset": "seqsleepnet",
30
+ "sequence_length": 200,
31
+ "max_epochs": 50,
32
+ "lr": 0.0001,
33
+ "weight_decay": 0.0001,
34
+ "batch_size": 8,
35
+ "fold": 0
36
+ },
37
+ "reference": "Phan et al. 2023 - L-SeqSleepNet: Whole-cycle Long Sequence Modelling for Automatic Sleep Staging"
38
+ }