dgl.distributed.load_partition

dgl.distributed.load_partition(part_config, part_id, load_feats=True, use_graphbolt=False)

从数据路径加载分区的数据。

分区数据包括分区的图结构、节点张量的字典、边张量的字典和一些元数据。分区可能包含HALO节点，这些节点是从其他分区复制的。然而，节点张量的字典仅包含属于本地分区的节点数据。同样，边张量仅包含属于本地分区的边数据。元数据包括全局图的信息（不是本地分区），包括节点数量、边数量以及全局图的节点分配。

该函数目前通过本地文件系统接口加载数据。

Parameters:

• part_config (str) – The path of the partition config file.

• part_id (int) – The partition ID.

• load_feats (bool, optional) – 是否加载节点/边特征。如果为False，返回的节点/边特征字典将为空。默认值：True。

• use_graphbolt (bool, optional) – 是否加载GraphBolt分区。默认值：False。

Returns:

• DGLGraph – 图分区结构。

• Dict[str, Tensor] – 节点特征。

• Dict[(str, str, str), Tensor] – 边特征。

• GraphPartitionBook – 图分区信息。

• str – 图名称

• List[str] – 节点类型

• List[(str, str, str)] – 边类型