Tensor Flow中的零丢失功能

Question

我正在检查Tensor Flow库中Zero-One损失的默认实现，以测量二进制分类的准确性。我找不到一个。有人可以指出我找到合适的资源。

Answer 1

如果您有输入：

• y_pred：您的预测。形状张量[batch_size, num_classes]，[0., 1.]
中的值
• y_true：真正的价值观。相同形状的张量，二进制值0或1

您可以使用以下方法计算损失和准确度：

y_pred_binary = tf.round(y_pred)
temp = tf.cast(tf.equal(y_pred_binary, y_true), tf.float32)
accuracy = tf.reduce_mean(temp, 1)
batch_loss = tf.reduce_sum(temp)

请记住，batch_loss没有渐变，因此您无法对丢失进行优化。相反，你可以使用sigmoid cross entropy。