Как получить точность 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, label = y)

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