/k-gnn

Source code for AAAI subimssion #3463.

Primary LanguageC++

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

This is the source code for the AAAI submission #3463: Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networks.

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.

Running examples

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