项目文件夹

文件

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