CAHC

From Representation to Clusters: A Contrastive Learning Approach for Attributed Hypergraph Clustering

Enviroment

torch==2.4.1 torch_cluster==1.6.3+pt24cu118 torch_scatter==2.1.2+pt24cu118 torch_sparse==0.6.18+pt24cu118 scikit-learn==1.7.1 pyg-lib==0.4.0+pt24cu118 Pygments==2.19.2

Basic Usage

python train.py --dataset cora --runs 5