https://github.com/shchur/overlapping-community-detection

Implementation of "Overlapping Community Detection with Graph Neural Networks"
https://github.com/shchur/overlapping-community-detection

community-detection geometric-deep-learning graph-neural-networks overlapping-community-detection pytorch

Last synced: 13 days ago
JSON representation

Implementation of "Overlapping Community Detection with Graph Neural Networks"

• Host: GitHub
• URL: https://github.com/shchur/overlapping-community-detection
• Owner: shchur
• License: mit
• Created: 2019-09-26T11:18:09.000Z (about 5 years ago)
• Default Branch: master
• Last Pushed: 2020-09-25T19:20:21.000Z (about 4 years ago)
• Last Synced: 2024-10-23T04:57:59.594Z (22 days ago)
• Topics: community-detection, geometric-deep-learning, graph-neural-networks, overlapping-community-detection, pytorch
• Language: Jupyter Notebook
• Homepage: http://www.daml.in.tum.de/nocd
• Size: 16.3 MB
• Stars: 154
• Watchers: 4
• Forks: 44
• Open Issues: 4
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Overlapping Community Detection with Graph Neural Networks

Pytorch implementation of the **Neural Overlapping Community Detection** method from

["Overlapping Community Detection with Graph Neural Networks"](http://www.kdd.in.tum.de/nocd).

## Usage

The main algorithm and other utilities are implemented in the `nocd` package that can be installed as

```bash

python setup.py install

```

A Jupyter notebook [interactive.ipynb](interactive.ipynb) contains the code for training the model and analyzing the results.

Experiments in the paper have been performed using an older TensorFlow version of the code that can be found 

[here](https://figshare.com/s/30894e4172505d5dc070).

## Requirements

```

numpy=1.16.4

pytorch=1.2.0

scipy=1.3.1

```

## Cite

Please cite our paper if you use the code or the datasets in your own work

```

@article{

    shchur2019overlapping,

    title={Overlapping Community Detection with Graph Neural Networks},

    author={Oleksandr Shchur and Stephan G\"{u}nnemann},

    journal={Deep Learning on Graphs Workshop, KDD},

    year={2019},

}

```