使用python在文本聚类中绘制单词

Question

我的聚类结果如下图所示。

是否有任何类似于 fvid_clusters 的库可以生成如下所示的图？ （使用 Python）

Answer 1

• 第 1 步：将单词向量化为向量（每个向量的大小为 n）
• 第 2 步：使用 kmean 将它们聚类到 k 个集群中
• 第 3 步：对第 1 步中生成的向量运行 PCA，并将它们减少到 2 维（例如 x、y）
• 第 4 步：在 (x, y) 位置绘制单词，其中 (x,y) 是 PCA 从其大小为 n 的全尺寸向量减少的维度。将与步骤 2 中识别的聚类对应的单词着色（每个聚类使用一种颜色）。

Answer 2

绘图与您展示的不完全一样，但我喜欢使用两个工具来探索文本聚类的结果（两者都有 Python 绑定）。

白蚁：http://vis.stanford.edu/papers/termite

UMAP：https://umap-learn.readthedocs.io/en/latest/

这是使用 NMF（来自我的论文）使用 BBC Sports dataset 执行的主题建模的结果。