使用PyG、NetworkX对Graph图数据进行可视化

PyG没有提供可视化的工具,所以我们需要将PyG的图数据 torch_geometric.data.data.Data 格式转换为 networkx.classes.graph.Graph 格式,然后利用 NetworkX 的工具进行可视化。

对图数据集进行可视化需要使用到 NetworkX 这个工具包,如果没有安装的需要提前安装一下,命令如下:

pip install networkx

1、为了测试,首先自定义创建一张图,7个节点,6条边,3个类别

在这里插入图片描述

# 创建一个图

# 定义了边的信息,第一个维度为起点,第二个位置为终点
edge_index = torch.tensor([[0

猜你喜欢

转载自blog.csdn.net/m0_47256162/article/details/128747683
今日推荐