dmlc--dgl
Giuseppe Futia
6c3dba867a
* Update prerequisites of README * dependencies for pytorch models * dependencies for mxnet models * minor
Community Detection with Graph Neural Networks (CDGNN)
Paper link: https://arxiv.org/abs/1705.08415
Author's code repo: https://github.com/joanbruna/GNN_community
This folder contains a DGL implementation of the CDGNN model.
Dependencies
- PyTorch 0.4.1+
- requests
pip install torch requests
How to run
An experiment on the Stochastic Block Model in default settings can be run with
python train.py
An experiment on the Stochastic Block Model in customized settings can be run with
python train.py --batch-size BATCH_SIZE --gpu GPU --n-communities N_COMMUNITIES \
--n-features N_FEATURES --n-graphs N_GRAPH --n-iterations N_ITERATIONS \
--n-layers N_LAYER --n-nodes N_NODE --model-path MODEL_PATH --radius RADIUS