项目文件夹

文件
hbsun2113 8c750170e8 [Model][Pytorch] GraphSAGE (#403)
* simple implemention of GraphSAGE

* update the abstract method and `torch.nn` modules are more utilized
2019-02-22 15:48:53 -05:00
..
2019-02-22 15:48:53 -05:00
2019-02-22 15:48:53 -05:00

Inductive Representation Learning on Large Graphs (GraphSAGE)

Requirements

  • requests

bash pip install requests

Results

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

python graphsage.py --dataset cora --gpu 0
  • cora: ~0.8470
  • citeseer: ~0.6870
  • pubmed: ~0.7730