dgl.data

dgl.data 包包含由 DGL 托管的数据集,以及用于从外部资源下载、处理、保存和加载数据的实用工具。

基类

DGLDataset

用于创建图数据集的基本DGL数据集。

CSVDataset

从CSV文件加载和解析图形数据的Dataset类。

节点预测数据集

用于节点分类/回归任务的数据集

SSTDataset

斯坦福情感树库数据集。

KarateClubDataset

用于节点分类的Karate Club数据集

CoraGraphDataset

Cora引用网络数据集。

CiteseerGraphDataset

Citeseer 引用网络数据集。

PubmedGraphDataset

Pubmed引文网络数据集。

CoraFullDataset

用于节点分类任务的CORA-Full数据集。

AIFBDataset

用于节点分类任务的AIFB数据集

MUTAGDataset

用于节点分类任务的MUTAG数据集

BGSDataset

用于节点分类任务的BGS数据集

AMDataset

AM 数据集。

AmazonCoBuyComputerDataset

AmazonCoBuy 数据集中用于节点分类任务的“计算机”部分。

AmazonCoBuyPhotoDataset

用于节点分类任务的AmazonCoBuy数据集。

CoauthorCSDataset

'计算机科学(CS)'是Coauthor数据集中用于节点分类任务的部分。

CoauthorPhysicsDataset

用于节点分类任务的Coauthor数据集中的“物理”部分。

PPIDataset

用于归纳节点分类的蛋白质-蛋白质相互作用数据集

RedditDataset

用于社区检测(节点分类)的Reddit数据集

SBMMixtureDataset

对称随机块模型混合

FraudDataset

欺诈节点预测数据集。

FraudYelpDataset

欺诈Yelp数据集

FraudAmazonDataset

欺诈亚马逊数据集

BAShapeDataset

BA-SHAPES dataset from GNNExplainer: Generating Explanations for Graph Neural Networks

BACommunityDataset

BA-COMMUNITY dataset from GNNExplainer: Generating Explanations for Graph Neural Networks

TreeCycleDataset

TREE-CYCLES dataset from GNNExplainer: Generating Explanations for Graph Neural Networks

TreeGridDataset

TREE-GRIDS dataset from GNNExplainer: Generating Explanations for Graph Neural Networks

WikiCSDataset

Wiki-CS 是一个基于维基百科的节点分类数据集,来自 Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks

FlickrDataset

用于节点分类的Flickr数据集来自GraphSAINT: 基于图采样的归纳学习方法

YelpDataset

用于节点分类的Yelp数据集来自GraphSAINT: 基于图采样的归纳学习方法

PATTERNDataset

用于图模式识别任务的PATTERN数据集。

CLUSTERDataset

用于半监督聚类任务的CLUSTER数据集。

ChameleonDataset

来自多尺度属性节点嵌入的变色龙维基百科页面网络,后来由Geom-GCN: 几何图卷积网络修改

SquirrelDataset

关于松鼠的维基百科页面网络来自多尺度属性节点嵌入,后来由Geom-GCN: 几何图卷积网络修改

ActorDataset

来自Social Influence Analysis in Large-scale Networks 的电影-导演-演员-编剧网络的仅演员诱导子图,由Geom-GCN: Geometric Graph Convolutional Networks 引入

CornellDataset

康奈尔子集的WebKB,后来由Geom-GCN: Geometric Graph Convolutional Networks修改

TexasDataset

德克萨斯州子集的WebKB,后来由Geom-GCN: Geometric Graph Convolutional Networks修改

WisconsinDataset

威斯康星州子集来自WebKB,后来由Geom-GCN: Geometric Graph Convolutional Networks修改

RomanEmpireDataset

来自论文《A Critical Look at the Evaluation of GNNs under Heterophily: Are We Really Making Progress? <https://arxiv.org/abs/2302.11640>'__ 的罗马帝国数据集。

AmazonRatingsDataset

来自论文《A Critical Look at the Evaluation of GNNs under Heterophily: Are We Really Making Progress? <https://arxiv.org/abs/2302.11640>》的Amazon-ratings数据集。

MinesweeperDataset

来自论文《A Critical Look at the Evaluation of GNNs under Heterophily: Are We Really Making Progress? <https://arxiv.org/abs/2302.11640>》的扫雷数据集。

TolokersDataset

来自论文《A Critical Look at the Evaluation of GNNs under Heterophily: Are We Really Making Progress? <https://arxiv.org/abs/2302.11640>》的Tolokers数据集。

QuestionsDataset

来自论文《对异质性下GNN评估的批判性审视:我们真的在取得进展吗?<https://arxiv.org/abs/2302.11640>'__ 的问题数据集。

MovieLensDataset

用于边缘预测任务的MovieLens数据集。

边预测数据集

用于边缘分类/回归和链接预测的数据集

FB15k237Dataset

FB15k237链接预测数据集。

FB15kDataset

FB15k链接预测数据集。

WN18Dataset

WN18 链接预测数据集。

BitcoinOTCDataset

用于欺诈检测的BitcoinOTC数据集

ICEWS18Dataset

用于时间图的ICEWS18数据集

GDELTDataset

基于事件的时间图的GDELT数据集

图预测数据集

用于图分类/回归任务的数据集

QM7bDataset

用于图属性预测(回归)的QM7b数据集

QM9Dataset

用于图属性预测(回归)的QM9数据集

QM9EdgeDataset

用于图属性预测(回归)的QM9Edge数据集

MiniGCDataset

合成图分类数据集类。

TUDataset

TUDataset 包含许多用于图分类的图核数据集。

LegacyTUDataset

LegacyTUDataset 包含许多用于图分类的图核数据集。

GINDataset

Dataset Class for How Powerful Are Graph Neural Networks?.

FakeNewsDataset

虚假新闻图分类数据集。

BA2MotifDataset

BA-2motifs dataset from Parameterized Explainer for Graph Neural Network

ZINCDataset

用于图回归任务的ZINC数据集。

MNISTSuperPixelDataset

用于图分类任务的MNIST超像素数据集。

CIFAR10SuperPixelDataset

CIFAR10超像素数据集用于图分类任务。

数据集适配器

AsNodePredDataset

将数据集重新用于标准的半监督传导节点预测任务。

AsLinkPredDataset

将数据集重新用于链接预测任务。

AsGraphPredDataset

将数据集重新用于标准的图属性预测任务。

Utilities

utils.get_download_dir

获取下载目录的绝对路径。

utils.download

下载给定的URL。

utils.check_sha1

检查文件内容的sha1哈希值是否与预期的哈希值匹配。

utils.extract_archive

提取归档文件。

utils.split_dataset

将数据集分割为训练集、验证集和测试集。

utils.load_labels

从文件加载标签字典

utils.save_info

将数据集相关信息保存到磁盘。

utils.load_info

从磁盘加载数据集相关信息。

utils.add_nodepred_split

将给定的数据集分割为训练集、验证集和测试集,用于传导式节点预测任务。

utils.mask_nodes_by_property

根据给定的节点属性,为具有分布偏移的节点分割提供分割掩码,如评估图模型在结构分布偏移下的鲁棒性和不确定性中所提出的。

utils.add_node_property_split

根据给定的节点属性创建一个具有分布偏移的节点分割,如评估图模型在结构分布偏移下的鲁棒性和不确定性中所提出的。

utils.Subset

在指定索引处数据集的子集