Go to file
William L Hamilton 8e0d053da2
Update encoders.py
2017-12-27 16:23:07 -06:00
cora Added Cora. 2017-12-19 14:20:38 -08:00
graphsage Update encoders.py 2017-12-27 16:23:07 -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).