dmlc--dgl
0a51dc5435
* fix for new ntype API for blocks * adding two new interfaces
Inductive Representation Learning on Large Graphs (GraphSAGE)
- Paper link: http://papers.nips.cc/paper/6703-inductive-representation-learning-on-large-graphs.pdf
- Author's code repo: https://github.com/williamleif/graphsage-simple. Note that the original code is simple reference implementation of GraphSAGE.
Requirements
- requests
bash pip install requests
Results
Run with following (available dataset: "cora", "citeseer", "pubmed")
python3 train_full.py --dataset cora --gpu 0
- cora: ~0.8330
- citeseer: ~0.7110
- pubmed: ~0.7830
Train w/ mini-batch sampling (on the Reddit dataset)
python3 train_sampling.py
Accuracy: 0.9504