HomeSort by relevance Sort by last modified time
    Searched refs:prediction_key (Results 26 - 32 of 32) sorted by null

12

  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
head_test.py 30 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
    [all...]
dnn_linear_combined_test.py 828 prediction_key='classes'),
832 prediction_key='classes'),
835 metric_fn=_my_metric_op, prediction_key='probabilities')
    [all...]
dnn_test.py 837 prediction_key='classes'),
841 prediction_key='classes'),
844 metric_fn=_my_metric_op, prediction_key='probabilities')
    [all...]
linear_test.py 392 prediction_key='classes'),
396 prediction_key='classes'),
399 metric_fn=_my_metric_op, prediction_key='probabilities')
411 # Tests the case where the prediction_key is neither "classes" nor
421 prediction_key='bad_type')
    [all...]
estimator.py 676 prediction_key=None,
700 prediction_key: The key for a tensor in the `predictions` dict (output
721 prediction_key=prediction_key,
    [all...]
estimator_input_test.py 306 prediction_key='class',
  /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
state_space_model_test.py 249 for prediction_key, combined_value in combined_predict.items():
250 self.assertAllClose(combined_value, split_predict[prediction_key])

Completed in 2491 milliseconds

12