Branch: master
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
README.md
gnn.py
train.py

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

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