我知道tf.estimator.evaluate()
返回一个包含eval_metrics
定义的字段的字典。因此,我们可以通过设置相应的字段使evaluate()
返回模型精度,并使用result["accuracy"]
获得精度。
但是,tf.estimator.predict()
返回一个生成器,如何以类似的方式获得预测精度?如果不能,是否还有其他方法可以使用tf.estimator
获得预测精度?
我确实想使用tf.estimator
,因为它与代码的其他部分一致。
谢谢!
答案 0 :(得分:0)
类似的东西可能会帮助您:tf.nn.softmax_cross_entropy_with_logits(logits = output,标签= y)
来源:https://datascience.stackexchange.com/questions/28968/get-the-probabilities-of-tensorflow