tf.contrib.predictor.from_contrib_estimator

tf.contrib.predictor.from_contrib_estimator(
    estimator,
    prediction_input_fn,
    input_alternative_key=None,
    output_alternative_key=None,
    graph=None,
    config=None
)

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

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

Args:

  • 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.

Returns:

An initialized Predictor.

Raises:

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