huihun/graphsage
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
William L Hamilton 2017-12-27 16:31:49 -06:00 committed by GitHub
parent 4583602eec
commit fcf5eb5d63
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
README.md
View File

@ -1,5 +1,18 @@
Basic reference pytorch implementation of GraphSAGE. # Reference PyTorch GraphSAGE Implementation
Forgive the messiness; it was coded in a rush. ### Author: William L. Hamilton
Basic reference PyTorch implementation of GraphSAGE.
This reference implementation is not as fast as the TensorFlow version for large graphs, but the code is easier to read and it performs better on small-graph benchmarks.
The code is also intended to be more extensible and easier to work with the the TensorFlow version.
Currently, only supervised versions of GraphSAGE-mean and GraphSAGE-GCN are implemented.
#### Requirements
pytorch >0.2 is required.
#### Running examples
Execute `python -m graphsage.model` to run the Cora example. 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. It assumes that CUDA is not being used, but modifying the run functions in `model.py` in the obvious way can change this.