tensorboard_test.py 文件源码

def accuracy(labels_placeholder, inference):
    with tf.name_scope('test'):
        correct_prediction = tf.equal(
            tf.arg_max(inference, 1), tf.argmax(labels_placeholder, 1)
        )
        acc = tf.reduce_mean(tf.cast(correct_prediction, 'float'))
        tf.scalar_summary('accuracy', acc)
        return acc