Skip to content

devsisters/neural-combinatorial-rl-tensorflow

Repository files navigation

Neural Combinatorial Optimization in Tensorflow

TensorFlow implementation of Neural Combinatorial Optimization with Reinforcement Learning.

model

(in progress)

Requirements

  • Python 2.7
  • tqdm
  • TensorFlow 0.12.1

Usage

To train a model:

$ python main.py --task=tsp20 --lr_start=0.001 --min_data_length=5 --max_data_length=20
$ python main.py --task=tsp50 --lr_start=0.001 --min_data_length=5 --max_data_length=50
$ python main.py --task=tsp100 --lr_start=0.0001 --min_data_length=5 --max_data_length=100

To train a model:

$ python main.py
$ tensorboard --logdir=logs --host=0.0.0.0

To test a model:

$ python main.py --is_train=False

Results

(in progress)

Author

Taehoon Kim / @carpedm20

Releases

No releases published

Packages

No packages published

Languages