如何在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中绘制此图吗？

谢谢。

- White

---

请查看此链接：https://discuss.pytorch.org/t/pytorch-geometric/44994 - Charlie Parker

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)

- Chao Shu

---

如何添加节点ID？ - Frank Xu