daunfamily / VRP_MHA Public

forked from Rintarooo/VRP_DRL_MHA

Notifications You must be signed in to change notification settings
Fork 1
Star 0

Attention learn to solve routing problems, TensorFlow2 (TF2), PyTorch, Capacitated Vehicle Routing Problem (CVRP), Transformer, Multi Head Attention, Deep Reinforcement Learning (DRL) (Rollout baseline, REINFORCE) (https://arxiv.org/pdf/1803.08475.pdf)

0 stars 38 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 58 Commits
PyTorch		PyTorch
TensorFlow2		TensorFlow2
LICENSE		LICENSE
README.md		README.md
makegif.py		makegif.py

Repository files navigation

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'

About

Attention learn to solve routing problems, TensorFlow2 (TF2), PyTorch, Capacitated Vehicle Routing Problem (CVRP), Transformer, Multi Head Attention, Deep Reinforcement Learning (DRL) (Rollout baseline, REINFORCE) (https://arxiv.org/pdf/1803.08475.pdf)

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%