/pinet2

Pytorch implementation of Pinet

Primary LanguagePython

PiNet

Pytorch implementation of Pinet also investigating further graph operators than in the original paper.

Work on this project is ongoing and is described in detail in the original paper.

If you use this work, please cite the following paper:

Meltzer, P., Mallea, M. D. G., & Bentley, P. J. (2019). PiNet: Attention Pooling for Graph Classification, NeurIPS 2019 Graph Representation Learning Workshop