dgl.load_graphs

dgl.load_graphs(filename, idx_list=None)

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

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

参数:

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

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

返回值:

• 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