Have a question? Connect with the community at the TensorFlow Forum Visit Forum


View source on GitHub

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

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.

An initialized Predictor.

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