params.json

{
  "train_dataset_size": 200,
  "validation_dataset_size": 100,
  "baseline_dataset_size": 100,
  "overfit_test": 0,

  "num_nodes": 30,
  "num_depots": 1,

  "sample_size": 5,
  "use_beam_search": true,
  "embedding_size": 128,
  "decoder_input_size": 514,
  "gradient_clipping": true,

  "num_neighbors_encoder": 20,
  "num_neighbors_action": 12,
  "num_movers": 1,
  "use_fleet_attention": false,

  "baseline_update_period": 5,
  "device": "cpu",

  "run_tests": true,
  "save_results": true,
  "learning_rate": 0.0001,
  "batch_size": 16,
  "test_batch_size": 16,
  "num_epochs": 50,

  "dataset_path": "https://huggingface.co/datasets/Cainiao-AI/LaDe/resolve/main/delivery/delivery_sh.csv"
}