在 Python 中聚类两个特征

Question

我有两个稀疏的 scipy 矩阵，title and paragraph，其维度分别为 (284,183) 和 (284,4195)。两个矩阵的每一行都是来自我的数据集一个实例的特征。我希望在没有预定义数量的集群的情况下对它们进行聚类，然后绘制它们。

我还有一个与每一行相关的数组Y。 (284,1)。一个类用 0 表示，另一个用 1 表示。我想用这个给点上色。如何使用 Python 做到这一点？

Answer 1

您可以使用skikit-learn Affinity propagation 或Mean-shift 库进行聚类。这些算法将输出一些集群和中心。使用Y 似乎是一个不同的问题，因为除非您导入一些其他库，否则您无法在 3D 平面上绘制多维点。