diff options
author | 2017-10-06 11:37:42 -0700 | |
---|---|---|
committer | 2017-10-06 11:53:53 -0700 | |
commit | 3110185270e93e0b6a3e82be9199febed1239602 (patch) | |
tree | b9f06efb708c1dbe672bd8ac1e36a5492875f812 /tensorflow/contrib/factorization/examples | |
parent | 7fceb8d879dd23a2fd15403d216367e5e8f52b56 (diff) |
Use the new Estimator.get_variable_value() method to get the kmeans cluster centers.
PiperOrigin-RevId: 171320755
Diffstat (limited to 'tensorflow/contrib/factorization/examples')
-rw-r--r-- | tensorflow/contrib/factorization/examples/mnist.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/factorization/examples/mnist.py b/tensorflow/contrib/factorization/examples/mnist.py index 9eefbccd4d..06a62db004 100644 --- a/tensorflow/contrib/factorization/examples/mnist.py +++ b/tensorflow/contrib/factorization/examples/mnist.py @@ -142,7 +142,7 @@ def inference(inp, num_clusters, hidden1_units, hidden2_units): # initial_clusters=tf.contrib.factorization.KMEANS_PLUS_PLUS_INIT, use_mini_batch=True) - (all_scores, _, clustering_scores, _, _, kmeans_init, + (all_scores, _, clustering_scores, _, kmeans_init, kmeans_training_op) = kmeans.training_graph() # Some heuristics to approximately whiten this output. all_scores = (all_scores[0] - 0.5) * 5 |