/GNNs-Baseline

A Pytorch based implementation of classical GNNs.

Primary LanguagePythonApache License 2.0Apache-2.0

GNNs-Baseline

A Pytorch based implementation of classical GNNs.
GNN基准测试模型.
Some components of this code are adapted from "GPRGNN" and "H2GCN".

Requirement

This project should be able to run without any modification after following packages installed.

pytorch
torch_geometric
networkx
torch-sparse

GNNs supported

H2GCN FAGCN APPNP GCN GAT ChebNet APPNP JKNet GPRGNN

Datasets supported

Downlaod by yourself, dataset_utils.py will help you.

cora, citeseer, pubmed, computers, photo, chameleon, squirrel, film, texas, cornell, wisconsin

Run experiment with GCN & Cora:

go to folder src

python train_model.py --RPMAX 2 \
        --net GCN \
        --train_rate 0.6 \
        --val_rate 0.2 \
        --dataset cora

Visualization of Confusion Matrix