tensorflow.contrib中的KMeans与KMeans聚类

时间:2018-07-26 09:10:44

标签: python tensorflow k-means

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

1 个答案:

答案 0 :(得分:1)

我正在使用sklearn进行此类工作,但通过文档了解这是我的理解:

如您随附的示例中所述,

KmeansClustering以文本形式print 'point:', point, 'is in cluster', cluster_index, 'centered at', center返回模型的结果。

Kmeans所做的事情完全相同,但是同时生成了模型的统计信息和图形。

我建议您同时尝试它们,看看自己的不同之处。