TensorFlow implementation of Pointer Networks.
(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