dmlc--dgl
GaiYu0
b420a5b5b3
* pretty printer * Conflicts: python/dgl/data/sbm.py * refined line_graph implementation * fix broken api calls * small fix to trigger CI * requested change
792 B
792 B
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.
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