微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决如何将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 举报,一经查实,本站将立刻删除。