Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ELG (Ensemble of Local and Global policies)

This repository is the code of the https://arxiv.org/abs/2308.14104, which ensembles a transferrable local policy to boost generalization. We provide the trained models to reproduce the test results in the paper.

Test ELG-POMO on VRPLIB

Under the ELG/CVRP folder, use the default settings in config.yml, run

python test_vrplib.py

You can choose the vrplib_set config from {X, XXL} to test on two different VRPLIB sets.

Test ELG-POMO on TSPLIB

Under the ELG/TSP folder, use the default settings in config.yml, and run

python test_tsplib.py

Train ELG-POMO on CVRP or TSP

First, generate the validation sets by

python generate_data.py

Modify the load_checkpoint config in config.yml to Null (i.e., load_checkpoint: ), and run

python train.py

About

Official implementation of IJCAI'24 paper "Towards Generalizable Neural Solvers for Vehicle Routing Problems via Ensemble with Transferrable Local Policy"

Topics

Resources

Stars

25 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages