tf.compat.v1.saved_model.classification_signature_def

View source on GitHub

Creates classification signature from given examples and predictions.

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.

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.

A classification-flavored signature_def.

ValueError If examples is None.