marc-match-ai / config.yaml
RvanB's picture
Implement gradio demo
a85bc9a
raw
history blame
528 Bytes
model:
batch_size: 512
features:
- title_tokenset
- title_agg
- author
- publisher
- pub_date
- pub_place
- pagination
hidden_sizes:
- 32
- 64
lr: 0.006
max_epochs: -1
optimizer: Adam
patience: 20
saved_models_dir: saved_models
test_processed_path: data/202303_goldfinch_set_1.1/processed/test_processed.csv
train_processed_path: data/202303_goldfinch_set_1.1/processed/train_processed.csv
val_processed_path: data/202303_goldfinch_set_1.1/processed/val_processed.csv
weight_decay: 0.0