Pytorch implementation of the Deep Variational Generative Overlapping Community Detection
The main algorithm and other utilities(metrics, evaluation .etc) are implemented in the embedding_cd
package that can be installed as
python setup.py install
The main analysis and visualization code are in Jupyter notebooks
run
pip install -r requirements.txt
first.
numpy>=1.16.4
pytorch>=1.2.0
scipy>=1.3.1
dgl>=0.42
...
Some modifications in DGL version:
- append
F.relu
after logits BatchNorm
after GCNLayeynocd.utils.l2_reg_loss(model, scale=args.weight_decay)
rather than setting weight_decay inAdam
dropout -> Relu -> BN