Karateclub：如何从图中提取类

Question

我试图从维基百科的图表中表示不同的节点

数据包含与鳄鱼相关的不同类别（动物、区域等），但我不知道这些信息在哪里

https://github.com/benedekrozemberczki/MUSAE

以下是基本图表：

from matplotlib import pyplot as plt
import networkx as nx
from karateclub import DeepWalk
import numpy as np

from karateclub import GraphReader
 
reader = GraphReader("wikipedia")

g = reader.get_graph()

target = reader.get_target()

features = reader.get_features()

plt.figure(figsize=(40,30))
nx.draw(g)
plt.show()

目标是通过优化图形中的颜色来可视化类

例如，使用DeepWalk：

model = DeepWalk(dimensions=2)
model.fit(g)
embedding = model.get_embedding()

plt.figure(figsize=(40,30))

plt.scatter(embedding[:,0],embedding[:,1])
plt.show()

但我没有任何颜色的类来清楚地比较结果，只有以下视图：

你知道怎么上这些课吗

我的目的是用一个清晰的可视化示例来证明图形优化算法（如DeepWalk和Graph2Vec）