Update README.md

This commit is contained in:
William L Hamilton 2018-03-10 10:33:12 -08:00 committed by GitHub
parent fcf5eb5d63
commit 146406c2c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,9 +2,9 @@
### 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.
Basic reference PyTorch implementation of [GraphSAGE](https://github.com/williamleif/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 (in terms of speed) on small-graph benchmarks.
The code is also intended to be simpler, more extensible, and easier to work with than the TensorFlow version.
Currently, only supervised versions of GraphSAGE-mean and GraphSAGE-GCN are implemented.