Go to file
William L Hamilton 4583602eec
Update model.py
2017-12-27 16:27:30 -06:00
cora Added Cora. 2017-12-19 14:20:38 -08:00
graphsage Update model.py 2017-12-27 16:27:30 -06:00
pubmed-data First commit. 2017-12-19 13:42:06 -08:00
.gitignore First commit. 2017-12-19 13:42:06 -08:00
README.md Create README.md 2017-12-19 16:32:14 -06:00

Basic reference pytorch implementation of GraphSAGE. Forgive the messiness; it was coded in a rush.

Execute python -m graphsage.model to run the Cora example. It assumes that CUDA is not being used, but modifying the run functions in model.py in the obvious way can change this. There is also a pubmed example (called via the run_pubmed function in model.py).