mirrors / dmlc--dgl

项目文件夹

dmlc--dgl
文件
0.4.x
T
Minjie Wang d876680a36 [Model][Perf] Improve sage sampling performance (#1364) 
* improve speed

* fix bugs

* upd reg test
2020-03-15 23:20:30 +08:00

849 B
原始文件 永久链接 Blame 文件历史

Inductive Representation Learning on Large Graphs (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

在新工单中引用 查看 Git Blame 复制永久链接