Skip to content

Latest commit

 

History

History
19 lines (15 loc) · 795 Bytes

README.md

File metadata and controls

19 lines (15 loc) · 795 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

Run Experiments

python cluster_gcn.py