diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-10-05 07:52:36 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-05 08:00:57 -0700 |
| commit | 7d9f8ffdcaf48968b137f7e785d04a689436449f (patch) | |
| tree | d677ef641b0da12208f5b7bdd738238eb88c7d45 /tensorflow/contrib/factorization/__init__.py | |
| parent | 6cf9ffeab4da4ad38bdf2afd803bf44cdc58d15d (diff) | |
Make a branch of the KMeans estimator that is ported to the core Estimator API.
PiperOrigin-RevId: 171152686
Diffstat (limited to 'tensorflow/contrib/factorization/__init__.py')
| -rw-r--r-- | tensorflow/contrib/factorization/__init__.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tensorflow/contrib/factorization/__init__.py b/tensorflow/contrib/factorization/__init__.py index 486c2ea933..6112c9d830 100644 --- a/tensorflow/contrib/factorization/__init__.py +++ b/tensorflow/contrib/factorization/__init__.py @@ -23,22 +23,24 @@ from tensorflow.contrib.factorization.python.ops.clustering_ops import * from tensorflow.contrib.factorization.python.ops.factorization_ops import * from tensorflow.contrib.factorization.python.ops.gmm import * from tensorflow.contrib.factorization.python.ops.gmm_ops import * +from tensorflow.contrib.factorization.python.ops.kmeans import * from tensorflow.contrib.factorization.python.ops.wals import * # pylint: enable=wildcard-import from tensorflow.python.util.all_util import remove_undocumented _allowed_symbols = [ - 'KMeans', 'COSINE_DISTANCE', - 'KMEANS_PLUS_PLUS_INIT', - 'RANDOM_INIT', - 'SQUARED_EUCLIDEAN_DISTANCE', - 'WALSModel', 'GMM', 'gmm', 'GmmAlgorithm', + 'KMeans', + 'KMEANS_PLUS_PLUS_INIT', + 'KMeansClustering', + 'RANDOM_INIT', + 'SQUARED_EUCLIDEAN_DISTANCE', 'WALSMatrixFactorization', + 'WALSModel', ] remove_undocumented(__name__, _allowed_symbols) |