yobibyte / pgn

Graph Nets in pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

sorting example

TODOs

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

Acknowledgements

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.

About

Graph Nets in pytorch

License:MIT License


Languages

Language:Python 100.0%