alarm_emb.yaml 912 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
---
# Dataset Stuff -------------------------------------------------
#

target_name: "Alarm"
model_path: models/alarm/embeddings
threshold_weeks: 8
threshold_training: 10

# Embedding Hyperparams --------------------------------------

train_ratio: 0.8
batch_size: 32
num_epochs: 10
verbose: False
network_layers: [128]
optimizer: "Adam"

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

features_to_normalize: ['BirthYear', 'Cluster', 'LoanPeriod', 'NumberAts']
features_to_scale: ['Gender_Male', 'Gender_Female', 'BirthYear',
                    'Cluster', 'LoanPeriod', 'NumberAts']

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

features: ['Gender_Male', 'Gender_Female', 'BirthYear',
           'Cluster', 'LoanPeriod']

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

use_real_ats_names: False
ats_resolution: 50