Go to file
William L Hamilton 664a8d633e
Create README.md
2017-12-19 16:32:14 -06:00
cora Added Cora. 2017-12-19 14:20:38 -08:00
graphsage Added Cora. 2017-12-19 14:20:30 -08: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 Added README. 2017-12-19 14:23:56 -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).