complete_emb.yaml 917 Bytes
Newer Older
thecml's avatar
thecml committed
1
2
3
4
5
6
---
# Dataset Stuff -------------------------------------------------
#

target_name: "Complete"
model_path: models/complete/embeddings
thecml's avatar
thecml committed
7
8
threshold_weeks: 8
threshold_training: 10
thecml's avatar
thecml committed
9

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

train_ratio: 0.8
batch_size: 32
num_epochs: 5
15
verbose: False
thecml's avatar
thecml committed
16
network_layers: [128]
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
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
36
ats_resolution: 50