Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
predictions = tf.nn.softmax(logits)
correct_prediction = tf.equal(tf.argmax(labels, 1), tf.argmax(predictions, 1))
accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment