如何获得tf.estimator.predict()的准确性?

时间:2020-02-06 02:44:42

标签: python tensorflow machine-learning tensorflow-estimator

我知道tf.estimator.evaluate()返回一个包含eval_metrics定义的字段的字典。因此,我们可以通过设置相应的字段使evaluate()返回模型精度,并使用result["accuracy"]获得精度。

但是,tf.estimator.predict()返回一个生成器,如何以类似的方式获得预测精度?如果不能,是否还有其他方法可以使用tf.estimator获得预测精度?

我确实想使用tf.estimator,因为它与代码的其他部分一致。

谢谢!

1 个答案:

答案 0 :(得分:0)

类似的东西可能会帮助您:tf.nn.softmax_cross_entropy_with_logits(logits = output,标签= y)

来源:https://datascience.stackexchange.com/questions/28968/get-the-probabilities-of-tensorflow