Tensorflow中Estimator中的自定义eval_metric_ops

Tensorflow中Estimator中的自定义eval_metric_ops

eval_metric_ops

需要按名称键入度量指标的字典。dict的值是调用度量函数的结果。您可以使用

tf.metrics

以下指标来实现您的指标功能：

 def metric_fn(labels, predict):    SST, update_op1 = tf.metrics.mean_squared_error(labels, tf.reduce_mean(labels))    SSE, update_op2 = tf.metrics.mean_squared_error(labels, predictions )    return tf.subtract(1.0, tf.div(SSE, SST)), tf.group(update_op1, update_op2))