dmlc--dgl
Hongzhi (Steve), Chen
23d09057e3
* [Misc] Black auto fix. * sort Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
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