jennyzhang0215/gaan

GaAN

The MXNet implementation of GaAN: Gated Attention Networks for Learning on Large and Spatiotemporal Graphs in UAI 2018.

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

Download datasets

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

python download_data.py --dataset ppi

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

Run experiments

The script is experiments/static_graph/sup_train_sample.py.

Citation

@inproceedings{zhang18,
  author    = {Jiani Zhang and Xingjian Shi and Junyuan Xie and Hao Ma and Irwin King and Dit{-}Yan Yeung},
  title     = {GaAN: Gated Attention Networks for Learning on Large and Spatiotemporal Graphs},
  booktitle = {Proceedings of the Thirty-Fourth Conference on Uncertainty in Artificial Intelligence},
  pages     = {339--349},
  year      = {2018}
}

gaan's Issues

AttributeError: module 'mxnet.symbol.contrib' has no attribute 'seg_pool'?

I download the "incubator-mxnet" in the same dir with "GaAN", and copy files as the "README.md". But when i run the "run.sh" in the dir of "cfg_ppi_sup", but it still not works. The problem is:

  File "/home/augf/GaAN/mxgraph/layers/aggregators.py", line 518, in hybrid_forward
    max_pool_data = F.contrib.seg_pool(data=max_pool_data,
AttributeError: module 'mxnet.symbol.contrib' has no attribute 'seg_pool'

And the "README.md" in GraphSampler, what's the meaning of "end points" , "ind_ptr" and "node_ids" and what's the advantage of using this format?
In fact, I want to test the accuary of 'cora' dataset in your model in cpu. Could you give me a python scripts for preprocessing the original dataset and runing the model?

what's the meaning of "incubator-mxnet/src/operator/contrib" in "segs_ops_cuda/README.md"?

Failed to recompile mxnet with custom seg_ops

I've followed the README to clone the mxnet and copied GaAN's custom operators. But when I run "make", seg_op.cc returned with serveral errors like:
.../incubator-mxnet/src/operator/contrib/seg_op.cc:371:18: error: ‘NodePtr’ in namespace ‘nnvm’ does not name a type; did you mean ‘NodeEntry’?

.../incubator-mxnet/src/operator/contrib/seg_op.cc:380:2: error: ‘struct mxnet::op::<lambda(int)>’ has no member named ‘add_argument’ 380 | .add_argument("data", "NDArray-or-Symbol", "The input data.")

Any possible solution?
gcc verion: 9.4.0
g++ version 9.4.0
mxnet version: linux_cu112.cmake
cuda version 11.2

brief previus steps:
git clone {mxnet}
cp {seq_op*} {mxnet}/src/contrib
mkdir build
cd build
cmake ..
make -j6

all previous built-in operators had complied fine.

ModuleNotFoundError: No module named 'mxgraph._graph_sampler'

Traceback (most recent call last):
  File "experiments/static_graph/sup_train_sample.py", line 4, in <module>
    from mxgraph.graph import *
  File "/home/augf/GaAN/mxgraph/graph.py", line 4, in <module>
    import mxgraph._graph_sampler as _graph_sampler
ModuleNotFoundError: No module named 'mxgraph._graph_sampler'

[Question] About USE_SHARPNESS option

Please tell me about the option USE_SHARPNESS. Thank you!

can not download the dataset ppi

when i try the command:

python download_data.py --dataset ppi

i get this problem:

urllib.error.URLError: <urlopen error [Errno 111] Connection refused