/GCN-NPEC

PyTorch implementation of GCN-NPEC in "Efficiently Solving the Practical Vehicle Routing Problem: A Novel Joint Learning Approach"

Primary LanguagePython

GCN-NPEC in PyTorch

A simple PyTorch implementation of Efficiently Solving the Practical Vehicle Routing Problem: A Novel Joint Learning Approach (KDD'20)

Dependencies

  • Python >= 3.6
  • PyTorch = 1.5
  • tqdm
  • scipy
  • numpy
  • plotly (only for plotting)
  • matplotlib (only for plotting)

Usage

Generate the pickle file containing hyper-parameter values by running the following command.

python config.py

You would see the pickle file in Pkl dir. now you can start training the model.

python train.py -p Pkl/***.pkl

Plot prediction of the trained model

python plot.py -p Weights/***.pt -b 128 -n 20

Reference