Вычисление пользовательских метрик с помощью tf.estimator.DNNRegressor в TensorFlow 1.10

Question

Как настроить tf.estimator.DNNRegressor для отчета о различных показателях, таких как RMSE и MAE при оценке?

(Можно задать один и тот же вопрос для tf.estimator.DNNClassifierи AUC метрика)

Примечание: Я знаю, что это должно быть сделано в tf.estimator.EstimatorSpec из model_fn() для пользовательского tf.estimator.Estimator, ноЯ не знаю, как применить это для tf.estimator.DNNRegressor.

Answer 1

вы можете использовать tf.contrib.estimator.add_metrics