项目文件夹

文件
Mufei Li 01e50626d3 [Example] Add TorchMetrics in README (#3913)
* Update README.md

* Update README.md

* Update README.md

Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2022-04-09 23:25:51 +08:00

795 B

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