Stay organized with collections Save and categorize content based on your preferences.

View source on GitHub

Constructs a Predictor from a tf.contrib.learn.Estimator.

estimator an instance of tf.contrib.learn.Estimator.
prediction_input_fn a function that takes no arguments and returns an instance of InputFnOps.
input_alternative_key Optional. Specify the input alternative used for prediction.
output_alternative_key Specify the output alternative used for prediction. Not needed for single-headed models but required for multi-headed models.
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 core Estimator instead of a contrib Estimator.