更新2020-11-23 Tensorflow中的Keras

Question

我是TensorFlow的新手。我正在使用自己的数据集进行二进制分类。但是我不知道如何计算准确度。有人可以帮我这么做吗？

我的分类器有5个卷积层，后面跟着2个完全连接的层。最终的FC层的输出维度为2，我使用过：

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

Answer 1

只需计算正确预测的百分比：

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

Answer 2

现在，您只需在metrics的{{1}}参数中指定即可即可。

这篇文章是从3.6年前开始的，当时tensorflow仍处于版本1中。现在Tensorflow.org建议使用Keras调用，您可以这样指定您想要的准确性：

model.compile

BOOM！当您运行“ model.fit”时，报告中的准确性很高。

如果您正在使用旧版的tensorflow或只是从头开始编写，@ Androbin会很好地解释它。