Skip to content

Latest commit

 

History

History
 
 

cluster_gcn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

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+
  • scikit-learn
  • TorchMetrics 0.11.4

Run Experiments

python cluster_gcn.py