dmlc--dgl
01e50626d3
* Update README.md * Update README.md * Update README.md Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
795 B
795 B
Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks
- Paper link: Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks
- Author's code repo: https://github.com/google-research/google-research/blob/master/cluster_gcn/.
This repo reproduce the reported speed and performance maximally on Reddit and PPI. However, the diag enhancement is not covered, as the GraphSage aggregator already achieves satisfying F1 score.
Dependencies
- Python 3.7+(for string formatting features)
- PyTorch 1.9.0+
- sklearn
- TorchMetrics
Run Experiments
python cluster_gcn.py