mirrors / dmlc--dgl

项目文件夹

文件

T

文件历史

Chao Ma f99725adbc all demo use python-3 (#555 )

2019-05-23 10:38:05 +08:00

gnn.py

[Model] SBM hotfix (#137 )

2018-11-27 11:58:03 -05:00

README.md

all demo use python-3 (#555 )

2019-05-23 10:38:05 +08:00

train.py

[Model] SBM hotfix (#137 )

2018-11-27 11:58:03 -05:00

README.md

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

python3 train.py

An experiment on the Stochastic Block Model in customized settings can be run with

python3 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