PtrNets
An implementation of Pointer Networks arXiv:1506.03134
Usage
To generate training data
For Convex Hull task:
>> python misc/convex_hull.py ch.pkl.gz
or for TSP task:
>> python misc/tsp.py tsp.pkl.gz
This will generate the training data, add the data path to ptrnets.py (or change the default path in data_util), then start training:
>> python ptrnets.py
References
- Oriol Vinyals, Meire Fortunato, Navdeep Jaitly, "Pointer Networks", arXiv:1506.03134 [stat.ML].