/PA-GCL

Primary LanguagePython

PA-GCL

Requirements

Install the following packages:

  • A C++ 11/14 compiler, e.g. gcc 4.8 or later
  • pytorch
  • lshknn
  • eigen 3
  • setuptools
  • 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