【dgl框架】dgl.metapath_reachable_graph函数解析

最新推荐文章于 2022-08-23 09:01:55 发布

一穷二白到年薪百万

最新推荐文章于 2022-08-23 09:01:55 发布

阅读量1.2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： # dgl 文章标签：机器学习人工智能

本文链接：https://blog.youkuaiyun.com/zfhsfdhdfajhsr/article/details/125665523

dgl 专栏收录该内容

11 篇文章

订阅专栏

本文介绍了DGL库中用于处理异构图神经网络的metapath_reachable_graph函数。该函数根据指定的元路径，构建一个新图，其中每个节点的邻居是可以通过元路径可达的节点。例如，给定元路径['AB', 'BA']，从A集合出发，新图表示了所有A节点通过AB和BA边可以到达的A节点。新图的边关系可以通过例子中的g.edges(order='eid')进行验证，展示了元路径在图结构中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 函数解析

如果用dgl学习异构图神经网络，就会遇到这个函数。这个函数的作用就是返回异构图中指定元路径(meta path)的metagraph。
看官方API的解释：

Return a graph where the successors of any node u are nodes reachable from u by the given metapath.

意思是：返回一个图，其中任何节点u的后续节点都是可以通过给定元路径从u访问的节点。

1.1 举个例子

这里指定的元路径为[‘AB’, ‘BA’]，经过dgl.metapath_reachable_graph函数返回的结果的含义就是：A集合中的节点经过元路径访问到的A集合节点，所构成的图（这个图是个方阵）。也可以理解为：A->B->A。

g = dgl.heterograph({
    ('A', 'AB', 'B'): ([0, 1, 2], [1, 2, 3]),
    ('B', 'BA', 'A'): ([1, 2, 3], [0, 1, 2])})
new_g = dgl.metapath_reachable_graph(g, ['AB', 'BA'])
new_g.edges(order='eid')
(tensor([0, 1, 2]), tensor([0, 1, 2]))