我们在训练分类模型时,需要输出模型预测的正确率用以评估,下面的代码片段可以实现这个功能。
# y_pred是模型的输出值,取值在[0,1]
# label是真实值,0或1
one = tf.ones_like(y_pred)
zero = tf.zeros_like(y_pred)
label_pred = tf.where(y_pred < 0.5, x=zero, y=on
我们在训练分类模型时,需要输出模型预测的正确率用以评估,下面的代码片段可以实现这个功能。
# y_pred是模型的输出值,取值在[0,1]
# label是真实值,0或1
one = tf.ones_like(y_pred)
zero = tf.zeros_like(y_pred)
label_pred = tf.where(y_pred < 0.5, x=zero, y=on