引文网络数据集PyG代码

PyG-citation-network

GNN模型在引文网络数据集上的代码，包括Cora、Citeseer、Pubmed、ogbn-arxiv

我的代码：https://github.com/ytchx1999/PyG-citation-network

参考代码：

• https://github.com/snap-stanford/ogb/blob/master/examples/nodeproppred/arxiv/gnn.py
• https://github.com/rusty1s/pytorch_geometric/blob/master/examples/ogbn_products_sage.py
• https://github.com/rusty1s/pytorch_geometric/blob/master/examples/reddit.py
• https://github.com/ytchx1999/PyG-GNN-Test

Cora、Citeseer、Pubmed

文件/文件夹	说明
full-citation.py	批处理： full-batch ；图数据表示方法： GS ；模型： GCN、GAT、SAGE、JKNet
mini-citation.py	批处理： mini-batch + 2-hop采样[10，10] ；图数据表示方法： GS ；模型： GraphSAGE

ogbn-arxiv

文件/文件夹	说明
full-ogbn.py	批处理： full-batch ；图数据表示方法： SpMM ；模型： GCN、GAT、SAGE、JKNet
mini-ogbn.py	批处理： mini-batch + 3-hop采样[15，10，5] ；图数据表示方法： SpMM ；模型： GraphSAGE