dgl.save_graphs

dgl.save_graphs(filename, g_list, labels=None, formats=None)[source]

将图以及可选的标签保存到文件。

除了保存到本地文件外，DGL 还支持直接将图写入 S3（通过提供 "s3://..." 路径）或 HDFS（通过提供 "hdfs://..." 路径）。

此函数将图结构以及节点/边特征以 DGL 自己的二进制格式保存到文件。对于图级别的特征，通过 labels 参数传递。

参数:

filename (str) – 用于存储图和标签的文件名。
g_list (list) – 要保存的图列表。
labels (dict[str, Tensor]) – 标签应为张量的字典，以字符串作为键。
formats (str 或 list[str]) – 按指定格式保存图。可以是 coo, csc 和 csr 的任意组合。如果未指定，则仅根据可用格式保存一种格式。如果多种格式可用，选择优先级从高到低为 coo, csc, csr。

示例

>>> import dgl
>>> import torch as th

创建 DGLGraph 对象并初始化节点和边特征。

>>> g1 = dgl.graph(([0, 1, 2], [1, 2, 3]))
>>> g2 = dgl.graph(([0, 2], [2, 3]))
>>> g2.edata["e"] = th.ones(2, 4)

将图保存到文件

>>> from dgl.data.utils import save_graphs
>>> graph_labels = {"glabel": th.tensor([0, 1])}
>>> save_graphs("./data.bin", [g1, g2], graph_labels)

另请参阅