From 664a8d633ed2f0fca9d2d739949a372821b3a06e Mon Sep 17 00:00:00 2001 From: William L Hamilton Date: Tue, 19 Dec 2017 16:32:14 -0600 Subject: [PATCH] Create README.md --- README.md | 6 ++++++ 1 file changed, 6 insertions(+) create mode 100644 README.md diff --git a/README.md b/README.md new file mode 100644 index 0000000..c5eeb7e --- /dev/null +++ b/README.md @@ -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).