python - 如何从 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