Name		Name	Last commit message	Last commit date
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

Repository files navigation

CVRP solver with Multi Head Attention

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

Desciption

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

Dependencies

Python > 3.5
TensorFlow > 2.0
tqdm
scipy
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'

About

TensorFlow 2.x, Vehicle Routing Problem (VRP), Deep Reinforcement Learning (DRL)(Rollout baseline, REINFORCE), Transformer, Multi Head Attention (MHA) (https://arxiv.org/pdf/1803.08475.pdf)

1 star

0 watching

Report repository

Releases

Packages

Contributors

Languages