TensorFlow系列:添加正确率(accuracy)统计算子
我们在训练分类模型时,需要输出模型预测的正确率用以评估,下面的代码片段可以实现这个功能。# y_pred是模型的输出值,取值在[0,1]# label是真实值,0或1one = tf.ones_like(y_pred)zero = tf.zeros_like(y_pred)label_pred = tf.where(y_pred < 0.5, x=zero, y=one)acc_op = tf.metrics.accuracy( labels=label, predictions=
原创
2020-05-12 22:27:07 ·
2360 阅读 ·
0 评论