Use streaming metrics to calculate proper values

raviqqe · Dec 25, 2016 · c2ba6d7 · c2ba6d7
1 parent 842fbd3
commit c2ba6d7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/examples/mnist_full/mnist_full.py b/examples/mnist_full/mnist_full.py
@@ -76,7 +76,8 @@ def model(image, number, mode):
     predictions = tf.argmax(h, axis=1)
     train_op = minimize(loss)
     eval_metric_ops = {
-        "accuracy": tf.reduce_mean(tf.to_float(tf.equal(predictions, number)))
+        "accuracy": tf.contrib.metrics.streaming_accuracy(predictions,
+                                                          number)[1],
     }
 
     if env("use_model_fn_ops"):

diff --git a/examples/mnist_simple/mnist_simple.py b/examples/mnist_simple/mnist_simple.py
@@ -40,7 +40,8 @@ def model(image, number):
     predictions = tf.argmax(h, axis=1)
 
     return predictions, loss, minimize(loss), {
-        "accuracy": tf.reduce_mean(tf.to_float(tf.equal(predictions, number)))
+        "accuracy": tf.contrib.metrics.streaming_accuracy(predictions,
+                                                          number)[1],
     }