TensorFlow 2.0 Beta is available Learn more

tf.estimator.export.PredictOutput

TensorFlow 2.0 version View source on GitHub

Class PredictOutput

Represents the output of a generic prediction head.

Inherits From: ExportOutput

Aliases:

  • Class tf.compat.v1.estimator.export.PredictOutput
  • Class tf.compat.v2.estimator.export.PredictOutput

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__

View source

__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.

Properties

outputs

Methods

as_signature_def

View source

as_signature_def(receiver_tensors)