tensorflow tf.estimator 打印AUC

貌似只能在eval的时候打印

auc,auc_op = tf.metrics.auc(labels=labels, predictions=tf.sigmoid(logits))

if mode == tf.estimator.ModeKeys.EVAL:

    eval_metrics = {"auc":(auc,auc_op)}

    output_spec = tf.estimator.EstimatorSpec(
                mode=mode,
                loss=total_loss,
                eval_metric_ops=eval_metrics)