项目文件夹

文件
Zihao Ye 9a0511c8e9 [NN] nn modules & examples update (#890)
* upd

* damn it

* fuck

* fuck pylint

* fudge

* remove some comments about MXNet

* upd

* upd

* damn it

* damn it

* fuck

* fuck

* upd

* upd

* pylint bastard

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd
2019-11-03 17:29:42 -05:00
..

Predict then Propagate: Graph Neural Networks meet Personalized PageRank (APPNP)

Dependencies

  • MXNET 1.5+
  • requests

bash pip install torch requests

Code

The folder contains an implementation of APPNP (appnp.py).

Results

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

DGLBACKEND=mxnet python3 train.py --dataset cora --gpu 0
  • cora: 0.8370 (paper: 0.850)
  • citeseer: 0.713 (paper: 0.757)
  • pubmed: 0.798 (paper: 0.797)

Experiments were done on dgl datasets (GCN settings) which are different from those used in the original implementation. (discrepancies are detailed in experimental section of the original paper)