Awesome
Contrastive Laplacian Eigenmaps (COLES)
Overview
This repo contains an example implementation of the Neurips 2021 paper: Contrastive Laplacian Eigenmaps. This code is based on SSGC (Simple Spectral Graph Convolution). In this code, we provide codes for Table 2 (contrastive classification) and node clustering experiments (Table 7). To prevent unnecessary issues, we submit the data along with the code. For reddit and ogb-arxiv, we do not provide the code because the corresponding datasets are too big. We also provide the log file for comparisons to help if the code cannot run correctly in reviewers' environment (any unknown issues with packages etc.)
This home repo contains the implementation for citation networks (Cora, Citeseer, and Pubmed, Cora Full).
Dependencies
Our implementation works with PyTorch>=1.0.0
Data
We provide the citation network datasets under data/
.
Usage
$ python train_ssgc_(dataset_name).py
$ python train_ssgc_(dataset_name)_clustering.py