Giters

chrsmrrs / k-gnn

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

deep-learningend-to-endgraph-algorithmsgraph-neural-networksgraphshigher-orderpytorchweisfeier-lemanweisfeiler-lehman

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

About

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

deep-learningend-to-endgraph-algorithmsgraph-neural-networksgraphshigher-orderpytorchweisfeier-lemanweisfeiler-lehman

Languages

Language:C++ 59.3%Language:Python 37.7%Language:C 3.1%