diff options
author | 2017-10-30 16:58:34 -0700 | |
---|---|---|
committer | 2017-10-30 17:05:41 -0700 | |
commit | 73fdaf0b560dd086d60a1e053affc5bfeed00097 (patch) | |
tree | 4870b5edcb9793a627baf8d223b9ed2d23f5e6c8 /tensorflow/contrib/summary | |
parent | 72be26dc821c536cc7c16740166cebb1c9fb3efa (diff) |
Summary-writing support for Evaluators.
PiperOrigin-RevId: 173971621
Diffstat (limited to 'tensorflow/contrib/summary')
-rw-r--r-- | tensorflow/contrib/summary/summary.py | 1 | ||||
-rw-r--r-- | tensorflow/contrib/summary/summary_ops.py | 7 |
2 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/contrib/summary/summary.py b/tensorflow/contrib/summary/summary.py index 89031caadc..ca82ea094c 100644 --- a/tensorflow/contrib/summary/summary.py +++ b/tensorflow/contrib/summary/summary.py @@ -29,6 +29,7 @@ from tensorflow.contrib.summary.summary_ops import all_summary_ops from tensorflow.contrib.summary.summary_ops import always_record_summaries from tensorflow.contrib.summary.summary_ops import audio from tensorflow.contrib.summary.summary_ops import create_summary_file_writer +from tensorflow.contrib.summary.summary_ops import eval_dir from tensorflow.contrib.summary.summary_ops import generic from tensorflow.contrib.summary.summary_ops import histogram from tensorflow.contrib.summary.summary_ops import image diff --git a/tensorflow/contrib/summary/summary_ops.py b/tensorflow/contrib/summary/summary_ops.py index 9c71bf7740..6028360732 100644 --- a/tensorflow/contrib/summary/summary_ops.py +++ b/tensorflow/contrib/summary/summary_ops.py @@ -19,6 +19,8 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +import os + from tensorflow.contrib.summary import gen_summary_ops from tensorflow.python.eager import context from tensorflow.python.framework import constant_op @@ -272,3 +274,8 @@ def audio(name, tensor, sample_rate, max_outputs, family=None): name=scope) return summary_writer_function(name, tensor, function, family=family) + + +def eval_dir(model_dir, name=None): + """Construct a logdir for an eval summary writer.""" + return os.path.join(model_dir, "eval" if not name else "eval_" + name) |