Class RegressionOutput
Inherits From: ExportOutput
Represents the output of a regression head.
__init__
__init__(value)
Constructor for RegressionOutput
.
Args:
value
: a floatTensor
giving the predicted values. Required.
Raises:
ValueError
: if the value is not aTensor
with dtype tf.float32.
Properties
value
Methods
tf.estimator.export.RegressionOutput.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.