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

Question

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

但是，tf.estimator.predict() 返回一个生成器，我怎样才能以类似的方式获得预测精度？如果我不能，有没有其他方法可以使用tf.estimator 获得预测准确性？

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

提前致谢！

Answer 1

这样的东西可能会对您有所帮助：tf.nn.softmax_cross_entropy_with_logits(logits=output, labels=y)

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