Wiki-CS：一个基于维基百科的图神经网络基准数据集
机器学习
2022-01-11 v2 社会与信息网络
机器学习
摘要
我们提出 Wiki-CS，一个源自维基百科、用于图神经网络基准测试的新数据集。该数据集由对应于计算机科学文章的节点组成，边基于超链接，并有代表该领域不同分支的 10 个类别。我们使用该数据集评估半监督节点分类和单关系链接预测模型。我们的实验表明，这些方法在一个具有不同于早期基准的结构属性的新领域上表现良好。该数据集及数据流水线与基准实验的实现已在 https://github.com/pmernyei/wiki-cs-dataset 公开可用。
引用
@article{arxiv.2007.02901,
title = {Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks},
author = {Péter Mernyei and Cătălina Cangea},
journal= {arXiv preprint arXiv:2007.02901},
year = {2022}
}
备注
Graph Representation Learning and Beyond workshop (ICML 2020); corrected incorrect metrics due to issue in experiment implementation