Constructs a Predictor from a tf.contrib.learn.Estimator.
tf.contrib.predictor.from_contrib_estimator(
estimator, prediction_input_fn, input_alternative_key=None,
output_alternative_key=None, graph=None, config=None
)
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.
|
View source on GitHub