tf.contrib.predictor.from_estimator

tf.contrib.predictor.from_estimator(
    estimator,
    serving_input_receiver_fn,
    output_key=None,
    graph=None,
    config=None
)

Defined in tensorflow/contrib/predictor/predictor_factories.py.

Constructs a Predictor from a tf.python.estimator.Estimator.

Args:

  • estimator: an instance of learn.python.estimator.Estimator.
  • serving_input_receiver_fn: a function that takes no arguments and returns an instance of ServingInputReceiver compatible with estimator.
  • output_key: Optional string specifying the export output to use. If None, then DEFAULT_SERVING_SIGNATURE_DEF_KEY is used.
  • graph: Optional. The Tensorflow graph in which prediction should be done.
  • config: ConfigProto proto used to configure the session.

Returns:

An initialized Predictor.

Raises:

  • TypeError: if estimator is a contrib Estimator instead of a core Estimator.