tf.saved_model.classification_signature_def

View source on GitHub

Creates classification signature from given examples and predictions.

Aliases:

• tf.compat.v1.saved_model.classification_signature_def
• tf.compat.v1.saved_model.signature_def_utils.classification_signature_def
• tf.saved_model.signature_def_utils.classification_signature_def

tf.saved_model.classification_signature_def(
    examples,
    classes,
    scores
)

This function produces signatures intended for use with the TensorFlow Serving Classify API (tensorflow_serving/apis/prediction_service.proto), and so constrains the input and output types to those allowed by TensorFlow Serving.

Args:

• examples: A string Tensor, expected to accept serialized tf.Examples.
• classes: A string Tensor. Note that the ClassificationResponse message requires that class labels are strings, not integers or anything else.
• scores: a float Tensor.

Returns:

A classification-flavored signature_def.

Raises:

• ValueError: If examples is None.