Branch: master
Find file History
giuseppefutia and ylfdq1118 [Tutorial] Update prerequisites of README (#380)
* Update prerequisites of README

* dependencies for pytorch models

* dependencies for mxnet models

* minor
Latest commit 6c3dba8 Feb 11, 2019
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
README.md [Tutorial] Update prerequisites of README (#380) Feb 12, 2019
gcn.py [Model] fix GCN (#305) Dec 16, 2018
gcn_spmv.py [Graph][Model] Cache adj & inc; MX GCN (#307) Dec 17, 2018

README.md

Graph Convolutional Networks (GCN)

Dependencies

  • PyTorch 0.4.1+
  • requests

bash pip install torch requests

Codes

The folder contains two implementations of GCN. gcn.py uses user-defined message and reduce functions. gcn_spmv.py uses DGL's builtin functions so SPMV optimization could be applied.

Results

Run with following (available dataset: "cora", "citeseer", "pubmed")

python gcn_spmv.py --dataset cora --gpu 0
  • cora: ~0.810 (0.79-0.83) (paper: 0.815)
  • citeseer: 0.707 (paper: 0.703)
  • pubmed: 0.792 (paper: 0.790)