dgl.load_graphs

dgl.load_graphs(filename, idx_list=None)[source]

从由save_graphs()保存的文件中加载图及其可选的标签。

除了从本地文件加载外，DGL还支持直接从S3（通过提供"s3://..."路径）或从HDFS（通过提供"hdfs://..."路径）加载图。

Parameters:

  • 文件名 (str) – 从中加载图形的文件名。

  • idx_list (list[int], optional) – 如果文件包含多个图，则为要加载的图的索引。 默认情况下是加载文件中存储的所有图。

Returns:

  • graph_list (list[DGLGraph]) – 加载的图。

  • labels (dict[str, Tensor]) – 存储在文件中的图标签。如果没有存储标签，字典为空。 无论是否提供了 idx_list 参数， 返回的字典始终包含所有图的标签。

示例

按照save_graphs()中的示例。

>>> from dgl.data.utils import load_graphs
>>> glist, label_dict = load_graphs("./data.bin") # glist will be [g1, g2]
>>> glist, label_dict = load_graphs("./data.bin", [0]) # glist will be [g1]

另请参阅

save_graphs