| View source on GitHub |
Represents the output of a regression head.
Inherits From: ExportOutput
tf.estimator.export.RegressionOutput(
value
)
| Args | |
|---|---|
value | a float Tensor giving the predicted values. Required. |
| Raises | |
|---|---|
ValueError | if the value is not a Tensor with dtype tf.float32. |
| Attributes | |
|---|---|
value | |
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. |
© 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/RegressionOutput