Represents the output of a supervised eval process.
Inherits From: ExportOutput
tf.estimator.export.EvalOutput(
loss=None, predictions=None, metrics=None
)
This class generates the appropriate signature def for exporting eval output by type-checking and wrapping loss, predictions, and metrics values.
| Args | |
|---|---|
loss | dict of Tensors or single Tensor representing calculated loss. |
predictions | dict of Tensors or single Tensor representing model predictions. |
metrics | Dict of metric results keyed by name. The values of the dict can be one of the following: (1) instance of Metric class. (2) (metric_value, update_op) tuples, or a single tuple. metric_value must be a Tensor, and update_op must be a Tensor or Op. |
| Raises | |
|---|---|
ValueError | if any of the outputs' dict keys are not strings or tuples of strings or the values are not Tensors (or Operations in the case of update_op). |
| Attributes | |
|---|---|
loss | |
metrics | |
predictions | |
as_signature_def
as_signature_def(
receiver_tensors
)
Generate a SignatureDef proto for inclusion in a MetaGraphDef.
The SignatureDef will specify outputs as described in this ExportOutput, and will use the provided receiver_tensors as inputs.
| Args | |
|---|---|
receiver_tensors | a Tensor, or a dict of string to Tensor, specifying input nodes that will be fed. |
| Class Variables | |
|---|---|
| LOSS_NAME | 'loss' |
| METRICS_NAME | 'metrics' |
| METRIC_UPDATE_SUFFIX | 'update_op' |
| METRIC_VALUE_SUFFIX | 'value' |
| PREDICTIONS_NAME | 'predictions' |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/estimator/export/EvalOutput