Unofficial Python implementation of the DNMF overlapping community detection algorithm
Project description
dnmf-python
Unofficial Python implementation of the Discrete Non-negative Matrix Factorization (DNMF) overlapping community detection algorithm
Paper
Ye, Fanghua, Chuan Chen, Zibin Zheng, Rong-Hua Li, and Jeffrey Xu Yu. 2019. “Discrete Overlapping Community Detection with Pseudo Supervision.” In 2019 IEEE International Conference on Data Mining (ICDM), 708–17. https://doi.org/10.1109/ICDM.2019.00081.
Official implementation in MATLAB at https://github.com/smartyfh/DNMF.
Requirements
python>=3.7.1
torch>=1.9.1
Quick start
-
To install the package run one of the two commands:
python -m pip install dnmf-python
(installation from PyPI)python setup.py install
(compile from source, if cloned the repository)
-
To run the algorithm, load the graph adjacency matrix into a
torch.FloatTensor
(for ex.A
), then call:from dnmf.DNMF import DNMF dnmf = DNMF() F = dnmf(A)
-
To run a quick test of the algorithm with an example graph, run
python test.py
from inside thesrc/dnmf/
directory
Config
The DNMF module supports the following hyperparameters as arguments:
alpha
: tradeoff parameter for the U-subproblembeta
: tradeoff parameter for the F-subproblemgamma
: regularization parameterk
: desired number of overlapping communitiesnum_outer_iter
: number of iterations for the outer loop (SDP iterations)num_inner_iter
: number of iterations for the inner loops (U and F subproblems)
How to cite
If you used dnmf-python
for work on your paper please use the following BibTeX entry to cite this software:
@misc{janchevski_dnmf_2021,
title = "dnmf-python",
author = "{Janchevski, Andrej}",
howpublished = "\url{https://github.com/Bani57/dnmf-python}",
year = 2021,
note = "Unofficial Python implementation of the Discrete Non-negative Matrix Factorization (DNMF) overlapping community detection algorithm"
}
Author
Andrej Janchevski
EPFL STI IEM LIONS
Lausanne, Switzerland
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for dnmf_python-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 51f79d44bc73eb6946ac9412a2b0d8be27e224a51e23fc8006e3a2d64d891fb2 |
|
MD5 | a6eafcc80eb03e00c1e5946c6d1133ef |
|
BLAKE2b-256 | 9a2c48f5aaf892f384cd60eddef884fce1e6976bdfbfc6edff23b3ec14b9496a |