fall_emb.yaml 1.68 KB
Newer Older
thecml's avatar
thecml committed
1
2
3
4
5
6
---
# Dataset Stuff -------------------------------------------------
#

target_name: "Fall"
model_path: models/fall/embeddings
7
fall_period_months: 6
thecml's avatar
thecml committed
8

9
# Embedding Hyperparams --------------------------------------
thecml's avatar
thecml committed
10
11
12

train_ratio: 0.8
batch_size: 32
13
14
num_epochs_ats: 10
num_epochs_ex: 5
15
verbose: False
thecml's avatar
thecml committed
16
network_layers: [128]
17
18
19
20
21
optimizer: "Adam"

# Settings for data loader -------------------------------------------------
#

22
23
24
25
26
27
28
29
30
31
32
features_to_normalize: ['BirthYear', 'Cluster', 'LoanPeriod', 'NumberSplit',
                        'NumberScreening', 'NumberWeeks', 'MeanEvaluation',
                        'NumberTraining', 'NumberTrainingWeek',
                        'TimeBetweenTraining', 'NumberWeeksNoTraining',
                        'Needs', 'Physics', 'NumberAts', 'NumberEx']
features_to_scale: ['Gender_Male', 'Gender_Female', 'BirthYear', 'Cluster',
                        'LoanPeriod', 'NumberSplit', 'NumberScreening',
                        'NumberWeeks', 'MeanEvaluation',
                        'NumberTraining', 'NumberTrainingWeek',
                        'TimeBetweenTraining', 'NumberWeeksNoTraining',
                        'Needs', 'Physics', 'NumberAts', 'NumberEx']
33
34
35
36

# Settings for data script -------------------------------------------------
#

37
38
39
40
features:  ['Gender_Male', 'Gender_Female', 'BirthYear', 'Cluster',
            'LoanPeriod', 'NumberSplit', 'NumberScreening', 'NumberWeeks',
            'MeanEvaluation', 'NumberTraining', 'NumberTrainingWeek',
            'TimeBetweenTraining', 'NumberWeeksNoTraining', 'Needs', 'Physics']
41
42
43
44
45

# Settings for dataset -------------------------------------------------
#

use_real_ats_names: False
46
47
48
ats_resolution: 50
ex_resolution: 9
risk_period_months: 6