-
Notifications
You must be signed in to change notification settings - Fork 5
/
my_experiment.yaml
executable file
·16 lines (16 loc) · 1.03 KB
/
my_experiment.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
dataset:
name : my_dataset
module : physioex.data.shhs.shhs # this is the module containing your own designed dataset
class: SHHS # this is the class name of your dataset, must extend PhysioExDataset (physioex.data.base)
experiment:
name : my_experiment
module : physioex.train.networks.chambon2018 # this is the module containing your own designed model
class: Chambon2018Net # this is the class name of your model, must extend SleepModule (physioex.train.networks.base)
module_config : # this is the argument passed to your model constructor
n_classes: 5 # mandatory
sfreq: 100
n_times: 3000
input_transform : raw # raw or xsleepnet supported
target_transform : # can be None if your model classifies sequences of epochs
module : physioex.train.networks.utils.target_transform # this is the module containing your own designed target transform function, can be the same as the model module
function : get_mid_label # in this case the sequence in output is transformed in a single label taking the middle value