Deep Learning for Graph in MXNet

Implements the Multi-head Graph Attention code in MXNet.

We only support python3!

Installation

Compile the MXNet operators by following the guide in seg_ops_cuda. Install the graph sampler by following the guide in GraphSampler.

python setup.py develop

You can download the datasets via the *download_data.py script. The usage is like

python download_data.py --dataset all

The --dataset hyperparameter can be 'cora', 'ppi', and 'reddit'.

The script is experiments/static_graph/sup_train_sample.py.