Neural Combinatorial Optimization in Tensorflow

TensorFlow implementation of Neural Combinatorial Optimization with Reinforcement Learning.

model

(in progress)

Requirements

Usage

To train a model:

$ python main.py --task=tsp --lr_start=0.001 --min_data_length=5 --max_data_length=20
$ python main.py --task=tsp --lr_start=0.001 --min_data_length=5 --max_data_length=50
$ python main.py --task=tsp --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