Skip to content

Files

Latest commit

a1f7498 · Apr 10, 2023

History

History
19 lines (15 loc) · 801 Bytes

README.md

File metadata and controls

19 lines (15 loc) · 801 Bytes

Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks

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 0.11.4

Run Experiments

python cluster_gcn.py