Home
last modified time | relevance | path

Searched refs:prediction_dict (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drnn_common.py141 prediction_dict, labels): argument
163 prediction_dict[prediction_key.PredictionKey.CLASSES], labels,
170 predictions=prediction_dict[prediction_key.PredictionKey.CLASSES],
283 prediction_dict = {}
296 prediction_dict[
307 prediction_dict[predictions_name] = predictions
308 return prediction_dict
Ddynamic_rnn_estimator.py297 prediction_dict = {
303 prediction_dict = {predictions_name: predictions}
304 return prediction_dict
356 prediction_dict): argument
377 for k, v in prediction_dict.items()
503 prediction_dict = rnn_common.multi_value_predictions(
509 prediction_dict = _single_value_predictions(
516 prediction_dict.update(state_dict)
521 problem_type, prediction_type, sequence_length, prediction_dict,
536 prediction_dict)
[all …]
Ddynamic_rnn_estimator_test.py408 prediction_dict = sequence_estimator.predict(
411 state_piece = prediction_dict[dynamic_rnn_estimator._get_state_name(i)]
475 prediction_dict = estimator.predict(
480 for (k, v) in prediction_dict.items()
483 return prediction_dict
629 prediction_dict = sequence_estimator.predict(
632 sorted(list(prediction_dict.keys())),
638 predictions = prediction_dict[prediction_key.PredictionKey.CLASSES]
639 probabilities = prediction_dict[
767 prediction_dict = sequence_estimator.predict(
[all …]
Dstate_saving_rnn_estimator.py502 prediction_dict = rnn_common.multi_value_predictions(
512 batch.length, prediction_dict, labels)
515 prediction_dict.update(state_dict)
528 predictions=prediction_dict,
Dstate_saving_rnn_estimator_test.py581 prediction_dict = sequence_estimator.predict(
584 sorted(list(prediction_dict.keys())),
589 predictions = prediction_dict[prediction_key.PredictionKey.CLASSES]
590 probabilities = prediction_dict[prediction_key.PredictionKey.PROBABILITIES]
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_test.py181 for prediction_dict in result_iter:
182 self.assertTrue("leaf_index" in prediction_dict)
183 self.assertTrue("logits" in prediction_dict)
341 for prediction_dict in result_iter:
342 self.assertTrue("classes" in prediction_dict)
368 for prediction_dict in result_iter:
369 self.assertTrue("classes" in prediction_dict)
395 for prediction_dict in result_iter:
396 self.assertTrue("classes" in prediction_dict)
423 for prediction_dict in result_iter:
[all …]
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
Dstate_space_model_test.py370 prediction_dict = random_model.predict(features={
378 predicted_mean = prediction_dict["mean"].eval()
379 predicted_covariance = prediction_dict["covariance"].eval()