tensorflow - 如何在TensorFlow中计算CNN的准确性

Question

我是TensorFlow的新手。我正在使用自己的数据集进行二进制分类。但是我不知道如何计算精度。谁能帮我做到这一点？

我的分类器有5个卷积层，其后是2个完全连接的层。最终的FC层的输出尺寸为2，我已经使用了该尺寸:

prob = tf.nn.softmax(classification_features, name="output")

Answer 1

只需计算正确预测的百分比即可:

prediction = tf.math.argmax(prob, axis=1)
equality = tf.math.equal(prediction, correct_answer)
accuracy = tf.math.reduce_mean(tf.cast(equality, tf.float32))