/k-gnn

Source code for our AAAI paper "Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networks".

Primary LanguageC++

Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networks

This is the source code for the AAAI 2019 paper Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networks (Preprint).

Installation

The code is built upon the PyTorch Geometric package, which needs to be installed before running the examples. Please follow its installation instructions.

Finally, run

python setup.py install

in the root directory of this repository. Tested with PyTorch Geometric 1.4.

Running examples

cd examples
python 1-2-3-proteins.py