conv-conv.py 文件源码

def correct_prediction(correct_pred=0):
    for i in range(PIXEL):
        correct_prediction = tf.equal(tf.argmax(conv_5[0][i], 1), tf.argmax(y[i], 1))
        correct_pred += correct_prediction
    accuracy = tf.reduce_mean(tf.cast(correct_pred, tf.float32))
    return accuracy