Skip to content

Files

Latest commit

997ca12 · Jul 27, 2022

History

History

citation_network

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Jul 27, 2022
Jul 27, 2022
Jul 27, 2022
Jul 27, 2022

Node Classification on Citation Networks

This example shows how to use modules defined in dgl.nn.pytorch.conv to do node classification on citation network datasets.

Datasets

  • Cora
  • Citeseer
  • Pubmed

Models

Usage

python run.py [--gpu GPU] --model MODEL_NAME --dataset DATASET_NAME [--self-loop]

The hyperparameters might not be the optimal, you could specify them manually in conf.py.