compliance_emb.yaml 304 Bytes
Newer Older
thecml's avatar
thecml committed
1
2
3
4
5
6
7
8
9
10
11
12
13
---
# Dataset Stuff -------------------------------------------------
#

target_name: "Compliance"
model_path: models/compliance/embeddings

# Training Hyperparams --------------------------------------

train_ratio: 0.8
batch_size: 32
num_epochs: 20
verbose: True
thecml's avatar
thecml committed
14
15
network_layers: [128]
optimizer: "Adam"