Home
last modified time | relevance | path

Searched refs:predict_input_fn (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest_test.py43 predict_input_fn = numpy_io.numpy_input_fn(
45 return train_input_fn, predict_input_fn
56 predict_input_fn = numpy_io.numpy_input_fn(
58 return train_input_fn, predict_input_fn
74 input_fn, predict_input_fn = _get_classification_input_fns()
81 predictions = list(classifier.predict(input_fn=predict_input_fn))
98 input_fn, predict_input_fn = _get_regression_input_fns()
104 predictions = list(regressor.predict(input_fn=predict_input_fn))
184 input_fn, predict_input_fn = _get_classification_input_fns()
192 predictions = list(est.predict(input_fn=predict_input_fn))
[all …]
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dstargan_estimator_test.py210 predict_input_fn, argument
243 predictions = np.array([x for x in est.predict(predict_input_fn)])
285 predict_input_fn = numpy_io.numpy_input_fn(
292 predict_input_fn = self._numpy_input_fn_wrapper(predict_input_fn,
295 predict_input_fn = estimator.stargan_prediction_input_fn_wrapper(
296 predict_input_fn)
301 predict_input_fn=predict_input_fn,
Dgan_estimator_test.py214 self, train_input_fn, eval_input_fn, predict_input_fn, prediction_size, argument
246 predictions = np.array([x for x in est.predict(predict_input_fn)])
266 predict_input_fn = numpy_io.numpy_input_fn(
274 predict_input_fn=predict_input_fn,
293 predict_input_fn = numpy_io.numpy_input_fn(
301 predict_input_fn=predict_input_fn,
350 predict_input_fn=_predict_input_fn,
Dtpu_gan_estimator_test.py155 self, train_input_fn, eval_input_fn, predict_input_fn, prediction_size, argument
194 est.predict(predict_input_fn)])
228 def predict_input_fn(params): function
240 predict_input_fn=predict_input_fn,
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Ddnn_test.py400 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
402 classifier.predict_classes(input_fn=predict_input_fn, as_iterable=True))
405 classifier.predict(input_fn=predict_input_fn, as_iterable=True))
445 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
448 input_fn=predict_input_fn, as_iterable=True))
451 classifier.predict(input_fn=predict_input_fn, as_iterable=True))
489 predict_input_fn = functools.partial(_input_fn_float_label, num_epochs=1)
492 input_fn=predict_input_fn, as_iterable=True))
496 input_fn=predict_input_fn, as_iterable=True))
500 input_fn=predict_input_fn, as_iterable=True))
[all …]
Ddebug_test.py295 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
296 predictions = list(classifier.predict_classes(input_fn=predict_input_fn))
322 predict_input_fn = functools.partial(_input_fn_float_label, num_epochs=1)
323 predictions = list(classifier.predict_classes(input_fn=predict_input_fn))
326 classifier.predict_proba(input_fn=predict_input_fn))
506 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
508 list(classifier.predict_classes(input_fn=predict_input_fn)))
551 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
552 predictions1 = classifier.predict_classes(input_fn=predict_input_fn)
559 predictions2 = classifier2.predict_classes(input_fn=predict_input_fn)
[all …]
Dlinear_test.py209 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
212 input_fn=predict_input_fn, as_iterable=True))
217 classifier.predict(input_fn=predict_input_fn, as_iterable=True))
405 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
407 input_fn=predict_input_fn)))
460 predict_input_fn = functools.partial(input_fn, num_epochs=1)
462 classifier.predict_proba(input_fn=predict_input_fn))
528 predict_input_fn = functools.partial(input_fn, num_epochs=1)
531 input_fn=predict_input_fn, as_iterable=True))
534 input_fn=predict_input_fn, as_iterable=True))
[all …]
Ddnn_linear_combined_test.py585 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
588 input_fn=predict_input_fn, as_iterable=True))
593 classifier.predict(input_fn=predict_input_fn, as_iterable=True))
840 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
842 input_fn=predict_input_fn)))
1413 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
1414 regressor.predict_scores(input_fn=predict_input_fn, as_iterable=True)
1451 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
1453 input_fn=predict_input_fn)))
1518 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
[all …]
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
Dstructural_ensemble_test.py123 predict_input_fn = input_pipeline.predict_continuation_input_fn(
126 regressor.predict(input_fn=predict_input_fn)
147 predict_input_fn = input_pipeline.predict_continuation_input_fn(
149 regressor.predict(input_fn=predict_input_fn)
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dsimple_estimator_example.py90 def predict_input_fn(): function
94 prediction_iterable = estimator.predict(input_fn=predict_input_fn)
/external/tensorflow/tensorflow/contrib/distribute/python/
Destimator_integration_test.py86 predict_input_fn = numpy_io.numpy_input_fn(
123 for x in estimator.predict(predict_input_fn)
Destimator_training_test.py241 def predict_input_fn(): function
248 for x in estimator.predict(predict_input_fn)
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Destimators_test.py91 predict_input_fn = input_pipeline.predict_continuation_input_fn(
98 ) = list(second_estimator.predict(input_fn=predict_input_fn))
/external/tensorflow/tensorflow/contrib/autograph/examples/notebooks/
Drnn_keras_estimator.ipynb989 "def predict_input_fn(color_name):\n",
1012 " input_fn=lambda:predict_input_fn(color_name))\n",