dmlc--dgl
Giuseppe Futia
6c3dba867a
* Update prerequisites of README * dependencies for pytorch models * dependencies for mxnet models * minor
34 行
912 B
Markdown
34 行
912 B
Markdown
Community Detection with Graph Neural Networks (CDGNN)
|
|
============
|
|
|
|
Paper link: [https://arxiv.org/abs/1705.08415](https://arxiv.org/abs/1705.08415)
|
|
|
|
Author's code repo: [https://github.com/joanbruna/GNN_community](https://github.com/joanbruna/GNN_community)
|
|
|
|
This folder contains a DGL implementation of the CDGNN model.
|
|
|
|
Dependencies
|
|
--------------
|
|
* PyTorch 0.4.1+
|
|
* requests
|
|
|
|
```bash
|
|
pip install torch requests
|
|
```
|
|
|
|
How to run
|
|
----------
|
|
|
|
An experiment on the Stochastic Block Model in default settings can be run with
|
|
|
|
```bash
|
|
python train.py
|
|
```
|
|
|
|
An experiment on the Stochastic Block Model in customized settings can be run with
|
|
```bash
|
|
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
|
|
```
|