Go to file
William L Hamilton b0bbc30be5
Delete README
2017-12-19 16:32:24 -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.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).