TensorFlow implementation of Neural Combinatorial Optimization with Reinforcement Learning.
(in progress)
- Python 2.7
- tqdm
- TensorFlow 0.12.1
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
(in progress)
Taehoon Kim / @carpedm20