graph - 有什么方法可以将“DGL.heterograph”转换为“torch_geometric”图？

Question

我正在使用 dgl 库，因为它很容易理解。但是我需要 torch_geometric 中的几个模块，但它们不支持 dgl 图。

有什么方法可以将 dgl 图更改为 torch_geometric 图？我的数据集是在 dgl 图形中构建的，当我加载数据集时，我会将它们更改为 torch_geometric 图形。

Answer 1

使用同networkx构图作为中间步骤，您可以使用以下方法：

1. dgl.to_homogenous创建同dgl构图实例
2. dgl.to_networkx将同构图导出到networkx图
3. torch_geometric.utils.from_networkxnetworkx将齐次图读入pytorch geometric
4. torch_geometric.data.to_heterogenous将同构图转换为异构图

据我所见，这两个包都没有针对异构图的单独读取器/写入器。因此，需要这些额外的步骤。另一种选择是手动导出张量和直接创建异构pytorch geometric图。