Skip to content
/ GBP Public

PyTorch implementation of "Scalable Graph Neural Networks via Bidirectional Propagation"

Notifications You must be signed in to change notification settings

chennnM/GBP

Repository files navigation

Scalable Graph Neural Networks via Bidirectional Propagation

This repository contains a PyTorch implementation of "Scalable Graph Neural Networks via Bidirectional Propagation".(http://arxiv.org/abs/2010.15421)

Requirements

Datasets

The data folder includes three benchmark datasets(Cora, Citeseer, Pubmed). Other datasets can be downloaded from PPI, Yelp, Amazon2M and Friendster. We also provide code to convert datasets to our format (in convert folder).

Compilation

make

Running the code

  • To replicate the transductive learning results (Cora, Citeseer, Pubmed), run the following script
sh transductive.sh
  • To replicate the inductive learning results (PPI, Yelp, Amazon2M), run the following script
sh inductive.sh
  • To replicate the inductive results of Friendster, run the following script
sh friendster.sh

Citation

@article{cwdlydw2020gbp,
  title = {Scalable Graph Neural Networks via Bidirectional Propagation},
  author = {Ming Chen, Zhewei Wei, Bolin Ding, Yaliang Li, Ye Yuan, Xiaoyong Du and Ji-Rong Wen},
  year = {2020},
  booktitle = {{NeurIPS}},
}

About

PyTorch implementation of "Scalable Graph Neural Networks via Bidirectional Propagation"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published