tf.estimator.export.ClassificationOutput

Class ClassificationOutput

Inherits From: ExportOutput

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

Represents the output of a classification head.

Either classes or scores or both must be set.

The classes Tensor must provide string labels, not integer class IDs.

If only classes is set, it is interpreted as providing top-k results in descending order.

If only scores is set, it is interpreted as providing a score for every class in order of class ID.

If both classes and scores are set, they are interpreted as zipped, so each score corresponds to the class at the same index. Clients should not depend on the order of the entries.

__init__

__init__(
    scores=None,
    classes=None
)

Constructor for ClassificationOutput.

Args:

  • scores: A float Tensor giving scores (sometimes but not always interpretable as probabilities) for each class. May be None, but only if classes is set. Interpretation varies-- see class doc.
  • classes: A string Tensor giving predicted class labels. May be None, but only if scores is set. Interpretation varies-- see class doc.

Raises:

  • ValueError: if neither classes nor scores is set, or one of them is not a Tensor with the correct dtype.

Properties

classes

scores

Methods

as_signature_def

as_signature_def(receiver_tensors)