tf.estimator.export.PredictOutput

Represents the output of a generic prediction head.

Inherits From: ExportOutput

A generic prediction need not be either a classification or a regression.

Named outputs must be provided as a dict from string to Tensor,

Args
outputs A Tensor or a dict of string to Tensor representing the predictions.
Raises
ValueError if the outputs is not dict, or any of its keys are not strings, or any of its values are not Tensors.
Attributes
outputs

Methods

as_signature_def

View source

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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/estimator/export/PredictOutput