tf.contrib.factorization.KMeans在TensorFlow 2.0中的等效API是什么？

Question

在Tensorflow 1.13.1中，可以使用KMeans alogithm使用tf.contrib.factorization.KMeans构建聚类模型，但是在升级到TensorFlow 2.0之后，我发现很难找到适合构建集群模型的API。我为K-均值tf.compat.v1.estimator.experimental.KMeans找到了一个估值器API。然而，它并不像前者那样对开发人员友好。我需要使用TensorFlow实现集群模型，但不确定哪个API合适。

Answer 1

旧的KMeans API被重构为遵循Estimator，而tf.contrib.factorization.KMeans甚至在迁移到Tensorflow 2.0之前就被废弃了。在1.15和2.x中，做k均值的“估计”方法是相同的。

如果您正在寻找更“友好”的方式，据我所知，您将留在例如科学知识学习。评估人员不会很快离开，所以熟悉他们并不是一个坏主意(尽管有一个学习曲线)。