load_labels

class dgl.data.utils.load_labels(filename)[source]

基础类:

从文件加载标签字典

Parameters:

文件名 (str) – 用于加载DGLGraphs的文件名

Returns:

labels – 存储在文件中的标签字典(如果没有存储标签,则返回空字典)

Return type:

dict

示例

按照 save_graphs 中的示例。

>>> from dgl.data.utils import load_labels
>>> label_dict = load_graphs("./data.bin")