有什么方法可以把'DGL.heterograph‘转换成'torch_geometric’图？

Question

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

有没有办法将dgl图更改为torch_geometric图？我的数据集是在dgl图中构建的，我将在加载数据集时将它们转换为torch_geometric图。

Answer 1

以齐次networkx图作为中间步骤，可以使用以下方法：

1. dgl.to_homogenous创建一个同质dgl图，instance
2. dgl.to_networkx将同质图导出到networkx graph
3. torch_geometric.utils.from_networkx，将同质networkx图读入pytorch geometric
4. torch_geometric.data.to_heterogenous，将同质图转换为异构图

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