Go to file
2017-12-19 14:23:56 -08: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

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).