与估计的API评价中保存直方图

Question

是否有可能使用估计API节省评估过程中的柱状图？

我找不到，因为估计API的解决方案不写下评价过程中的任何总结，我只能加标量到求值指标。

Answer 1

您可以使用SummarySaverHook

eval_hooks = []

eval_summary_hook = tf.train.SummarySaverHook(
                    save_steps=1,
                    output_dir='model_dir',
                    summary_op=tf.summary.histogram(logits.name, logits))

eval_hooks.append(eval_summary_hook)

return tf.estimator.EstimatorSpec(mode=mode,
                                          loss=loss,
                                          eval_metric_ops=eval_metric_ops,
                                          evaluation_hooks=evaluation_hooks
                                          )