TUDataset:用于图学习的一组基准数据集
机器学习
2020-07-20 v1 神经与进化计算
机器学习
摘要
近来,人们对图数据的(监督)学习,尤其是使用图神经网络的兴趣日益增长。然而,有意义基准数据集与标准化评估程序的发展滞后,从而阻碍了该领域的进展。为此,我们引入了用于图分类与回归的 TUDataset。该集合包含来自广泛应用、规模各异的 120 多个数据集。我们提供了基于 Python 的数据加载器、核方法与图神经网络基线实现以及评估工具。在此,我们概述了这些数据集、标准化评估程序,并给出了基线实验。所有数据集可在 www.graphlearning.io 获取。实验可从 www.github.com/chrsmrrs/tudataset 的代码完全复现。
引用
@article{arxiv.2007.08663,
title = {TUDataset: A collection of benchmark datasets for learning with graphs},
author = {Christopher Morris and Nils M. Kriege and Franka Bause and Kristian Kersting and Petra Mutzel and Marion Neumann},
journal= {arXiv preprint arXiv:2007.08663},
year = {2020}
}
备注
ICML 2020 workshop "Graph Representation Learning and Beyond"