python - 如何可视化 DGL 数据集中的图表？

Question

以下代码片段来自教程 https://cnvrg.io/graph-neural-networks/ 。如何可视化数据集中的图表？如果可能的话，使用诸如 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