pgn

PGN (read as pagan) is a library to build graph networks in pytorch. If you love tensorflow, you can find the original implementation here.

Example

Run python3 sort.py in the `examples' folder to run a sorting toy-task. You should get the results similar to the picture below.

TODOs

• [ ] Update documentation.
• [ ] Add other examples, e.g. TSP.

Acknowledgements

• Wendelin Boehmer for useful discussions and trying it out.
• Matthias Fey for his great pytorch-scatter library.

Disclaimer

This library is highly experimental and designed for educational purposes. For me it was a way to understand Graph Networks.

pgn works with python3 only.