JhuoW / PE-GCL

Code for IJCAI 2022 paper PE-GCL

PE-GCL

Requirements

Install the following packages:

• pytorch
• lshknn
• scikit-learn
• networkx
• numpy==1.20.3
• torch_geometric==1.6.3
• tqdm==4.50.2
• PyYAML==5.4.1
• scikit_learn==0.24.2

Basic Usage

Node Classification

$ python run_simple_random_split.py --dataset Cora   # node classification
$ python run_simple_random_split.py --dataset CS