如何从DGL的数据集中可视化图形？

Question

以下代码片段来自教程https://cnvrg.io/graph-neural-networks/。如何从dataset中可视化图形？如果可能的话，使用像matplotlib这样的东西。

import dgl
import torch
import torch.nn as nn
import torch.nn.functional as F
import dgl.data
dataset = dgl.data.CoraGraphDataset()
g = dataset[0]

Answer 1

import dgl.data
import matplotlib.pyplot as plt
import networkx as nx

dataset = dgl.data.CoraGraphDataset()
g = dataset[0]
options = {
    'node_color': 'black',
    'node_size': 20,
    'width': 1,
}
G = dgl.to_networkx(g)
plt.figure(figsize=[15,7])
nx.draw(G, **options)

这是一个很大的图表，所以你可能需要调整大小(节点、宽度、图形等)。以下是一些有用的链接：

• https://networkx.org/documentation/stable/tutorial.html

• https://docs.dgl.ai/en/0.5.x/generated/dgl.to_networkx.html#dgl.to_networkx

• https://docs.dgl.ai/en/0.5.x/api/python/dgl.data.html