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