W3cubDocs

/TensorFlow Python

tf.estimator.export.PredictOutput

Class PredictOutput

Inherits From: ExportOutput

Defined in tensorflow/python/estimator/export/export_output.py.

Represents the output of a generic prediction head.

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

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

Properties

outputs

Methods

__init__

__init__(outputs)

Constructor for PredictOutput.

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.

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.

© 2018 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/api_docs/python/tf/estimator/export/PredictOutput