Cluster GCN implementation in pyg?

[Anarion-zuo]

Hi! I'm new here.

I'm looking for implementations of the Cluster GCN, and found this. In the paper, the model first cluster nodes, then forward node representation in a GNN layer. I don't see the clustering operation in this. Wonder where I can find it.

[EdisonLeeeee]

You can check out these examples:

• https://github.com/pyg-team/pytorch_geometric/blob/master/examples/cluster_gcn_reddit.py
• https://github.com/pyg-team/pytorch_geometric/blob/master/examples/cluster_gcn_ppi.py

The first step (graph partition/clustering) is implemented using ClusterData.