Skip to content

yobibyte/pgn

Repository files navigation

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages