From 6a103519566aeb2e74870862a421ad40bd3d67e7 Mon Sep 17 00:00:00 2001 From: Yelrose <270018958@qq.com> Date: Thu, 30 Jul 2020 15:14:45 +0800 Subject: [PATCH] add gcnii --- examples/citation_benchmark/README.md | 48 +++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) diff --git a/examples/citation_benchmark/README.md b/examples/citation_benchmark/README.md index 4313ec0..82fa0b2 100644 --- a/examples/citation_benchmark/README.md +++ b/examples/citation_benchmark/README.md @@ -1 +1,49 @@ # Easy Paper Reproduction for Citation Network (Cora/Pubmed/Citeseer) + + + +This page tries to reproduce all the **Graph Neural Network** paper for Citation Network (Cora/Pubmed/Citeseer), which is the **Hello world** dataset (**small** and **fast**) for graph neural networks. But it's very hard to achieve very high performance. + + + +All datasets are runned with public split of **semi-supervised** settings. + + + +How to run the experiments? + + + +```shell +# Device choose +export CUDA_VISIBLE_DEVICES=0 +# GCN +python train.py --conf config/gcn.yaml --use_cuda --dataset cora +python train.py --conf config/gcn.yaml --use_cuda --dataset pubmed +python train.py --conf config/gcn.yaml --use_cuda --dataset citeseer + + +# GAT +python train.py --conf config/gat.yaml --use_cuda --dataset cora +python train.py --conf config/gat.yaml --use_cuda --dataset pubmed +python train.py --conf config/gat.yaml --use_cuda --dataset citeseer + + +# SGC (Slow version) +python train.py --conf config/sgc.yaml --use_cuda --dataset cora +python train.py --conf config/sgc.yaml --use_cuda --dataset pubmed +python train.py --conf config/sgc.yaml --use_cuda --dataset citeseer + +# APPNP +python train.py --conf config/appnp.yaml --use_cuda --dataset cora +python train.py --conf config/appnp.yaml --use_cuda --dataset pubmed +python train.py --conf config/appnp.yaml --use_cuda --dataset citeseer + +# GCNII (The original code use 1500 epochs.) +python train.py --conf config/appnp.yaml --use_cuda --dataset cora --epoch 1500 +python train.py --conf config/appnp.yaml --use_cuda --dataset pubmed --epoch 1500 +python train.py --conf config/appnp.yaml --use_cuda --dataset citeseer --epoch 1500 +``` + + + -- GitLab