/GraphSAGE

GraphSAGE implemented in PyTorch.

Primary LanguagePython

GraphSAGE

GraphSAGE in PyTorch.

Usage

All use the PPI dataset by default.

Run the desired file as,

# Supervised graphsage
python graphsage.py

# Unsupervised graphsage
python graphsage_unsup.py