tf.estimator.export.build_parsing_serving_input_receiver_fn

tf.estimator.export.build_parsing_serving_input_receiver_fn(
    feature_spec,
    default_batch_size=None
)

Build a serving_input_receiver_fn expecting fed tf.Examples.

Creates a serving_input_receiver_fn that expects a serialized tf.Example fed into a string placeholder. The function parses the tf.Example according to the provided feature_spec, and returns all parsed Tensors as features.

Args:

  • feature_spec: a dict of string to VarLenFeature/FixedLenFeature.
  • default_batch_size: the number of query examples expected per batch. Leave unset for variable batch size (recommended).

Returns:

A serving_input_receiver_fn suitable for use in serving.