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

Question

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

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

Answer 1

图并不完全像您显示的那样，但是我想使用两种工具来探究文本聚类的结果（两个都有针对Python的绑定）。

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

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

这是NMF使用BBC Sports dataset使用NMF进行主题建模的结果。

Answer 2

• 第1步：将单词向量化为向量（每个大小为n）
• 第2步：使用kmean将它们聚类为k个聚类
• 第3步：对第1步中生成的向量运行PCA，并将其缩小为2维（例如x，y）
• 第4步：在位置（x，y）上绘制单词，其中（x，y）是PCA从大小为n的全尺寸向量中减小的维数。为与第2步中确定的聚类对应的单词上色（每个聚类使用一种颜色）。