CPGCL: Cluster-Perceptive Graph Contrastive Learning for Community Detection

Code implementation of paper "Cluster-Perceptive Graph Contrastive Learning for Community Detection". Accepted by ICASSP202.

Requirements

Install the dependencies: pip install -r requirements.txt.

First, unzip the dataset, which is saved in .txt file format:

unzip data.zip

Then run main.py to train the model (taking AMAP as an example):

python main.py --dataset amap --epoch 500 --lr 7e-4 --beta 0.4

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
README.md		README.md
data.zip		data.zip
main.py		main.py
module.py		module.py
opt.py		opt.py
requirements.txt		requirements.txt
train.py		train.py
utils.py		utils.py