PiGCL
December 20, 2023 ยท View on GitHub
The official source code for A New Mechanism for Eliminating Implicit Conflict in Graph Contrastive
Part of code is referenced from Deep Graph Contrastive Representation Learning
Environment Setup
- torch 1.13.0
- torch-geometric 2.3.0
- scikit-learn 1.2.1
- numpy 1.23.5
You can install the dependencies using the following command:
pip install -r requirements.txt
Usage
You can follow these commands to train and test a model with a specific dataset:
./run_script_Cora.sh
./run_script_CiteSeer.sh
./run_script_PubMed.sh
./run_script_CS.sh
./run_script_Photo.sh
./run_script_Computers.sh