Pointer Networks
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
pointer
README.md

README.md

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].