Added README.

jianchaoji · Dec 19, 2017 · 1966d21 · 1966d21
1 parent 127147f
commit 1966d21
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/README b/README
@@ -0,0 +1,6 @@
+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).