Graph neural networks for community detection (GNN4CD)

Graph Neural Networks and Line Graph Neural Networks for community detection in graphs, as described in the paper Supervised community detection with graph neural networks by Zhengdao Chen, Lisha Li and Joan Bruna, which appeared in ICLR 2019.

The implementation in the main branch uses Python (3.7) with NumPy (1.18.1) and PyTorch (1.4.0). A previou version compatible with Python 3.6 and PyTorch 0.3.1 can be found in the branch pytorch_0.3.1. The code is partially adapted from the code here.

Running script_5SBM_gnn.sh and script_5SBM_lgnn.sh will perform the experiments of community detection for 5-class dissortive stochastic block models using a graph neural network (GNN) and a line graph neural network (LGNN), respectively.