python - tensorflow.contrib 中的 KMeans 与 KMeansClustering

Question

根据 tensorflow 的文档，有两个类与 Kmeans 算法相关。其中之一是 tensorflow.contrib.factorization.KMeans 另一个是tensorflow.contrib.factorization.KMeansClustering .该文档指出前者用于为 kmeans 聚类创建图形，但我实际上并没有得到它。我想对 MNIST 数据集使用 kmeans 来聚类数字然后对它们进行分类。我的代码应该使用哪个类？

Answer 1

我正在使用 sklearn 来完成这类工作，但通过文档，这是我的理解:

如您附加的示例中所示，KmeansClustering 以文本形式返回模型的结果:print 'point:', point, 'is in cluster', cluster_index, 'centered at', center。

Kmeans 做完全相同的事情，但同时生成模型的统计数据和图表。

我建议您都尝试一下，看看自己的区别。