config.yaml

model:
  # Inputs features
  features:
    - title_tokenset
    - title_agg
    - author
    - publisher
    - pub_date
    - pub_place
    - pagination
  # Size of hidden layers
  hidden_sizes:
    - 32
    - 64
    
  # Training
  batch_size: 512
  weight_decay: 0.0
  max_epochs: -1

  # Disable early stopping with -1
  patience: 20
  
  lr: 0.006
  optimizer: Adam
  saved_models_dir: saved_models

  # Paths to dataset splits
  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