Class PredictOutput
Inherits From: ExportOutput
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
,
__init__
__init__(outputs)
Constructor for PredictOutput.
Args:
outputs
: ATensor
or a dict of string toTensor
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 notTensor
s.
Properties
outputs
Methods
tf.estimator.export.PredictOutput.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
: aTensor
, or a dict of string toTensor
, specifying input nodes that will be fed.