在TensorFlow 1.10中使用tf.estimator.DNNRegressor计算自定义指标

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