图数据集Planetoid御三家:Cora,CiteSeer,PubMed详解

简介: 图数据集Planetoid御三家:Cora,CiteSeer,PubMed详解

Cora,CiteSeer,PubMed应该是GNN学习过程中没有人能逃得过的三大关口,在数据集上的意义可能相当于GCN,GraphSAGE,GAT之于空域GNN的意义。

本文旨在介绍御三家数据集,包括其基本信息和使用方式。

您可以根据您所需数据使用情况来观看本文。譬如若您只想在PyG建模时调用该数据集,可以仅观看第一节;如果您想了解如何将原始数据处理成类似御三家数据集的格式,可以仅观看第三节。

Enjoy!


本文将首先介绍数据集的基本情况,接下来以集成度从高到底的顺序进行撰写数据集的调取和使用方式:

第2节,从PyG的数据集调用接口开始介绍数据集的使用方式,并在这一节介绍三个数据集的基本信息(毕竟集成度高,数据很好用)。

第3节,介绍Planetoid数据(PyG数据来源)的使用和处理方式。

第4节,介绍从原始数据来源下载并处理数据的方式,包括如何将这种类似的原始数据处理成类Planetoid数据的方式。

第5节,最后介绍另一个集成度较高、但是流行度较低的CogDL中的数据集接口及使用方式。


1. 数据集基本情况介绍


Cora,CiteSeer,PubMed是三个文献引用网络数据集。

都是一张无向图,节点代表文件(论文文献),边代表引用关系。

标签在节点上,多分类任务。


节点特征是word vector,每一元素都是0-1二元变量,描述每个单词在paper中是否存在。统计时经过了stemming1和去除停用词,并去掉了小于10篇文档中出现过的单词。


1.1 Cora数据集

2708个节点,5429条边。标签共7个类别。数据集的特征维度是1433维。


Cora数据集的节点是机器学习论文,论文主题(标签)分为以下七类之一:

Case_Based基于案例

Genetic_Algorithms遗传算法

Neural_Networks神经网络

Probabilistic_Methods概率方法

Reinforcement_Learning强化学习

Rule_Learning规则学习

Theory理论

论文的选择方式是,在最终语料库中,每篇论文引用或被至少一篇其他论文引用(即没有孤立点存在)。


1.2 SiteSeer数据集

3312个节点,4723条边。标签共7个类别。数据集的特征维度是3703维。


论文主题(标签)分为以下七类之一:

Agents

AI

DB

IR

ML

HCI

论文的选择方式是,在最终语料库中,每篇论文引用或被至少一篇其他论文引用(即没有孤立点存在)。


1.3 PubMed

略。


2. PyG数据集的Planetoid调用接口


torch_geometric.datasets.Planetoid官方文档

略。


PyG所使用的御三家数据来源就是这篇论文:Revisiting Semi-Supervised Learning with Graph Embeddings. Zhilin Yang, William W. Cohen, Ruslan Salakhutdinov. ICML 2016.。对此的详细介绍见本文第二节。


3. Planetoid官方数据处理方式


kimiyoung/planetoid: Semi-supervised learning with graph embeddings

略。


4. 原始数据处理介绍


数据集来源:Datasets | LINQS

Cora数据集下载地址:https://linqs-data.soe.ucsc.edu/public/lbc/cora.tgz

CiteSeer数据集下载地址:https://linqs-data.soe.ucsc.edu/public/lbc/citeseer.tgz

PubMed数据集下载地址:https://linqs-data.soe.ucsc.edu/public/Pubmed-Diabetes.tgz

Linux可以通过wget下载文件,并用tar解压。

tgz压缩包中都含有README文件,介绍数据情况。

Cora压缩包里的README文件里面说数据集来源是这个网站:www.research.whizbang.com/data,我打不开这个网站就很离谱……

略。


5. CogDL的数据集调用接口


CogDL官网(中文版):CogDL工具包

THUDM/cogdl: CogDL: An Extensive Toolkit for Deep Learning on Graphs

cogdl/gat.py at master · THUDM/cogdl

略。

相关文章
|
机器学习/深度学习 图计算 图形学
同构图、异构图、属性图、非显式图
同构图(Homogeneous Graph)、异构图(Heterogeneous Graph)、属性图(Property Graph)和非显式图(Graph Constructed from Non-relational Data)。 (1)同构图:
4043 0
同构图、异构图、属性图、非显式图
|
机器学习/深度学习 编解码 算法
英文论文(sci)解读复现:基于YOLOv5的自然场景下苹果叶片病害实时检测
英文论文(sci)解读复现:基于YOLOv5的自然场景下苹果叶片病害实时检测
1099 0
|
编译器
overleaf 参考文献引用,创建引用目录.bib文件,在文档中引用参考文献,生成参考文献列表
overleaf 参考文献引用,创建引用目录.bib文件,在文档中引用参考文献,生成参考文献列表
11513 0
|
3月前
|
安全 数据安全/隐私保护 开发者
AxureRP-Setup安装教程简单步骤Mac版(附安装包)
Axure RP是专业原型设计工具,助产品经理在编码前快速构建网页/App界面与交互流程,直观呈现产品形态。本文详解Mac端安装步骤:下载dmg、拖入应用程序、绕过macOS安全限制完成安装。(239字)
|
自然语言处理 算法 数据挖掘
【数据挖掘】十大算法之PageRank连接分析算法
文章介绍了PageRank算法的基本概念和数学模型,包括如何通过一阶马尔科夫链定义随机游走模型以及如何计算网页的重要性评分,并提供了PageRank迭代算法的具体步骤。
1228 1
|
弹性计算 分布式计算 网络协议
聊聊复杂网络环境下hdfs的BlockMissingException异常|参数dfs.client.use.datanode.hostname
企业真实的网络环境是复杂多变的,在复杂的网络环境中部署并使用 hadoop 时,如果服务端的配置或客户端的使用不当,就可能会遇见各种问题。
聊聊复杂网络环境下hdfs的BlockMissingException异常|参数dfs.client.use.datanode.hostname
|
存储 人工智能 编译器
【AI系统】算子手工优化
本文深入探讨了手写算子调度的关键因素及高性能算子库的介绍,通过计算分析指标和 RoofLine 模型评估计算与访存瓶颈,提出了循环、指令、存储三大优化策略,并介绍了 TVM 和 Triton 两种 DSL 开发算子的方法及其在实际应用中的表现。
1271 2
【AI系统】算子手工优化
|
关系型数据库 Linux 数据库
PostgreSQL 入门指南:安装、配置与基本命令
本文从零开始,详细介绍如何在 Windows、Linux 和 macOS 上安装和配置 PostgreSQL,涵盖30+个实操代码示例。内容包括安装步骤、配置远程访问和用户权限、基础数据库操作命令(如创建表、插入和查询数据),以及常见问题的解决方案。通过学习,你将掌握 PostgreSQL 的基本使用方法,并为后续深入学习打下坚实基础。
14312 1

热门文章

最新文章