Missed TensorFlow World? Check out the recap. Learn more

tf.compat.v1.saved_model.classification_signature_def

View source on GitHub

Creates classification signature from given examples and predictions.

Aliases:

  • tf.compat.v1.saved_model.signature_def_utils.classification_signature_def
tf.compat.v1.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.