A PyTorch implementation of GraphSAGE
This package contains a PyTorch implementation of GraphSAGE. Currently, only supervised versions of GraphSAGE-mean, GraphSAGE-GCN, GraphSAGE-maxpool and GraphSAGE-meanpool are implemented.
Authors of this code package:
Bin Yu
Environment settings
- python>=3.6.8
- pytorch>=1.0.0
Basic Usage
Example Usage
To run the supervised model on Cuda:
python train.py
