Pytorch Implementation of the CompleNet 2018 paper "Modularity Optimization as a Training Criterion for Graph Neural Networks".

Usage:

python run.py

Parameters:

--dataset 
Specify dataset to use. Possible values are 'cora', 'citeseer' or 'pubmed'

--weight
Specify weight value of community structure preserving term, valid values from 0.0 to 1.0

--n_label_per_class
Set number of labels per class to sample for training set, valid values range from 1 to 20