PyG(PyTorch Geometric)是一个用于图神经网络的Python库,提供了丰富的图数据集用于图分析和图神经网络的研究。这些内置的图数据集包含了各种不同领域的真实和合成图数据,可以帮助研究人员和开发者快速构建和测试他们的图神经网络模型。
下面是一些PyG内置的常见图数据集及其用途:
-
CORA(引文网络数据集):
CORA是一个引文网络数据集,用于文献分类任务。它包含了2708篇科学文献,每篇文献都有1433个词汇特征,并属于以下七个类别之一:物理学、计算机科学、数学、统计学、社会学、生物学和医学。研究人员可以使用CORA数据集来开发和评估图神经网络在文献分类方面的性能。from torch_geometric.datasets import Planetoid dataset = Planetoid(root='data/CORA