Skip to content

Latest commit

 

History

History
47 lines (29 loc) · 1.12 KB

README.md

File metadata and controls

47 lines (29 loc) · 1.12 KB

CVRP solver with Multi Heads Attention

TensorFlow2 and PyTorch implementation of ATTENTION, LEARN TO SOLVE ROUTING PROBLEMS!(https://arxiv.org/pdf/1803.08475.pdf)

Description

Slide Share -- CVRP solver with Multi Heads Attention --

Dependencies

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

Usage

First generate the pickle file contaning hyperparameter values by running the following command.

python config.py

then, train the model.

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

Plot prediction of the pretrained model

python plot.py -p './Weights/***.h5'