search

chrsmrrs / k-gnn

Source code for our AAAI paper "Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networks".
183 stars 45 forks source link
deep-learning end-to-end graph-algorithms graph-neural-networks graphs higher-order pytorch weisfeier-leman weisfeiler-lehman

readme

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