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 | |