TensorFlow implementation of Pointer Networks.
(in progress)
- Python 2.7
- tqdm
- TensorFlow 0.12.1
To train a model:
$ python main.py --task=tsp20 --min_length=5 --max_data_length=20 # download dataset from the paper $ python main.py --task=tsp20 --min_length=5 --max_data_length=10 # generate dataset itself 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