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

target_name: "Compliance"
model_path: models/compliance/embeddings
7
use_real_ats_names: False
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: 5
verbose: True
thecml's avatar
thecml committed
15
network_layers: [128]
16
metrics: ['accuracy']
17
18
19
20
21
optimizer: "Adam"

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

thecml's avatar
thecml committed
22
23
features_to_normalize: ['BirthYear', 'LoanPeriod', 'NumberAts']
features_to_scale: ['Gender', 'BirthYear', 'LoanPeriod', 'NumberAts']
24
25
26
27

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

thecml's avatar
thecml committed
28
features: ['Gender', 'BirthYear', 'LoanPeriod']