如何在Python中可视化torch_geometric图？

Question

让我们考虑一个例子，我有以下坐标格式的邻接矩阵：

> edge_index.numpy() = array([[    0,     1,     0,   3,   2],
                              [    1,     0,     3,   2,   1]], dtype=int64)

这意味着节点 0 链接到节点 1，反之亦然，节点 0 链接到 3 等...

您知道如何使用 nx.draw() 在 networkx 中绘制此图吗？

谢谢。

Answer 1

import networkx as nx

edge_index = torch.tensor([[0, 1, 1, 2],
                           [1, 0, 2, 1]], dtype=torch.long)
x = torch.tensor([[-1], [0], [1]], dtype=torch.float)

data = torch_geometric.data.Data(x=x, edge_index=edge_index)
g = torch_geometric.utils.to_networkx(data, to_undirected=True)
nx.draw(g)