config.yaml 326 Bytes
Newer Older
1
data_path: ../data/processed/
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

model_names: {
    'CNN': 'CNN_V3',
    'RNN': 'RNN_V2',
    'LR': 'LR',
    'RF': 'RF',
}

train:
    budget: 50
    repeat: 1
    epochs: 15

feature_dimension:
    ARF:
        4.0 : 4143
        12.0: 4912

    Shock:
        4.0 : 4620
        12.0: 5597

    mortality:
        48.0: 7508