Skip to content

Latest commit

 

History

History
50 lines (38 loc) · 1.48 KB

README.md

File metadata and controls

50 lines (38 loc) · 1.48 KB

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}},
}