- 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.
- Python 3.7+(for string formatting features)
- PyTorch 1.9.0+
- scikit-learn
- TorchMetrics 0.11.4
python cluster_gcn.py