数据集端点

class graphdatascience.procedure_surface.api.catalog.dataset_endpoints.DatasetEndpoints

load_cora(graph_name: str = 'cora', undirected: bool = False) → GraphV2

一个已引入的引用网络。

参数:

• graph_name (str) – 要创建的图的名称

• undirected (bool) – 图是否应为无向

返回:

该图的句柄。

返回类型:

GraphV2

load_imdb(graph_name: str = 'imdb', undirected: bool = True) → GraphV2

一个用于基准测试节点分类或链接预测模型的异构图。

该图包含演员（Actors）、导演（Directors）和电影（Movies）（以及未分类电影 UnclassifiedMovies）作为节点，并包含演员与他们出演的电影之间的关系，以及导演与他们执导的电影之间的关系。

参数:

• graph_name (str) – 要创建的图的名称

• undirected (bool) – 图是否应为无向

返回:

该图的句柄。

返回类型:

GraphV2

load_karate_club(graph_name: str = 'karate_club', undirected: bool = False) → GraphV2

一个由 http://konect.cc/networks/ucidata-zachary/[Zachary 引入的社交网络。

参数:

• graph_name (str) – 要创建的图的名称

• undirected (bool) – 图是否应为无向

返回:

该图的句柄。

返回类型:

GraphV2

load_lastfm(graph_name: str = 'lastfm', undirected: bool = True) → GraphV2

一个用于基准测试链接预测模型的异构图。原始数据来自 http://www.lastfm.com/[LastFM。

该图包含用户（User）和艺术家（Artists）作为节点，每个节点都有一个对应原始数据中 ID 的 rawId 字段。例如，可使用 rawId 回溯到 HetRec’11 的 .dat 文件以检查艺术家名称。

参数:

• graph_name (str) – 要创建的图的名称

• undirected (bool) – 图是否应为无向

返回:

该图的句柄。

返回类型:

GraphV2

property networkx

用于将 networkx 图加载到图目录的便捷包装器。

返回类型:

NXLoader

property ogbl: OGBLLoader

用于链接属性预测的数据集。

property ogbn: OGBNLoader

用于节点属性预测的数据集。