如何将pytorch几何图形转换为Networkx Multigraph?

问题描述

我有一个加权图存储在Data对象中,我想将此图转换为Networkx MultiGraph。这是我尝试过的:

data = Data(x=x,edge_index=edge_index)
print(edge_index).

.torch.Size([3,237])

转化

第一次审判

from torch_geometric.utils import to_networkx,from_networkx
G=to_networkx(data,to_undirected=True)

ValueError:太多值无法解包(预期2)

第二次审判

G = nx.MultiGraph()
edges = data.edge_index.numpy()
edge_list = []
for i in range(data.num_edges):
    edge_list.append(tuple(edges[:,i]))
G.add_edges_from(edge_list)
data.G = G
adj=nx.adjacency_matrix(G)
print('here the adj is',adj.shape)

结果=(18,18)预期(23,23)

我还能做什么?非常感谢您的投入

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)