CVRP solver with Multi Head Attention

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

Desciption

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'

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
Pkl		Pkl
Weights		Weights
__pycache__		__pycache__
LICENSE		LICENSE
README.md		README.md
config.py		config.py
data.py		data.py
decoder.py		decoder.py
encoder.py		encoder.py
env.py		env.py
layers.py		layers.py
model.py		model.py
plot.py		plot.py
rollout.py		rollout.py
train.py		train.py