| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| metric_spec.py | 243 returns a prediction with key "prediction_key".
245 In order to compute the accuracy of the "prediction_key" prediction, we
250 prediction_key="prediction_key",
263 prediction_key="prediction_key",
280 prediction_key="prediction_key",
289 prediction_key=None,
304 prediction_key: The key for a tensor in the `predictions` dict (outpu
327 def prediction_key(self): member in class:MetricSpec
[all...] |
| metric_spec_test.py | 61 metric_fn=fn, prediction_key="p1", label_key="l1", weight_key="f2")
212 spec = MetricSpec(metric_fn=fn, prediction_key="p1", label_key="l1")
229 "MetricSpec without specified prediction_key requires predictions"
242 spec = MetricSpec(metric_fn=_fn, prediction_key="p1", weight_key="f2")
276 spec = MetricSpec(metric_fn=_fn, prediction_key="p1", weight_key="f2")
291 metric_fn=_fn, prediction_key="p1", label_key="l1", weight_key="f2")
294 "MetricSpec with prediction_key specified requires predictions dict"):
307 metric_fn=_fn, prediction_key="p1", label_key="l1", weight_key="f2")
319 prediction_key="my_prediction",
334 prediction_key="my_prediction"
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| rnn_common.py | 24 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
31 # removed and replaced with values from `prediction_key.PredictionKey`. The key
33 # `prediction_key.PredictionKey.SCORES` for regression and
34 # `prediction_key.PredictionKey.CLASSES` for classification. The key
36 # `prediction_key.PredictionKey.PROBABILITIES`.
158 prediction_dict[prediction_key.PredictionKey.CLASSES], labels,
165 predictions=prediction_dict[prediction_key.PredictionKey.CLASSES],
248 containing single entry with key `prediction_key.PredictionKey.CLASSES` for
250 `prediction_key.PredictionKey.SCORE` for `problem_type`
254 `prediction_key.PredictionKey.PROBABILITIES`. Th
[all...] |
| debug.py | 53 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
212 key = prediction_key.PredictionKey.CLASSES
229 key = prediction_key.PredictionKey.PROBABILITIES
322 key = prediction_key.PredictionKey.SCORES
|
| dynamic_rnn_estimator_test.py | 31 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
193 {prediction_key.PredictionKey.CLASSES: True,
194 prediction_key.PredictionKey.PROBABILITIES: True,
198 {prediction_key.PredictionKey.CLASSES: True,
199 prediction_key.PredictionKey.PROBABILITIES: True})}),
203 {prediction_key.PredictionKey.SCORES: True,
208 {prediction_key.PredictionKey.SCORES: True})}),
212 {prediction_key.PredictionKey.CLASSES: True,
213 prediction_key.PredictionKey.PROBABILITIES: True,
491 pred_all_at_once[prediction_key.PredictionKey.CLASSES
[all...] |
| svm.py | 27 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
158 key = prediction_key.PredictionKey.CLASSES
175 key = prediction_key.PredictionKey.PROBABILITIES
|
| model_fn.py | 29 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
217 scores = output_tensors.get(prediction_key.PredictionKey.SCORES)
219 scores = output_tensors.get(prediction_key.PredictionKey.PROBABILITIES)
223 classes = output_tensors.get(prediction_key.PredictionKey.CLASSES)
|
| dnn.py | 34 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
448 key = prediction_key.PredictionKey.CLASSES
483 key = prediction_key.PredictionKey.PROBABILITIES
516 prediction_key=prediction_key.PredictionKey.PROBABILITIES,
677 custom_metrics[(key, prediction_key.PredictionKey.SCORES)] = metric
760 key = prediction_key.PredictionKey.SCORES
791 prediction_key=prediction_key.PredictionKey.SCORES,
[all...] |
| head.py | 29 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
759 key = prediction_key.PredictionKey.SCORES
[all...] |
| linear.py | 33 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
557 key = prediction_key.PredictionKey.CLASSES
588 key = prediction_key.PredictionKey.PROBABILITIES
620 prediction_key=prediction_key.PredictionKey.PROBABILITIES,
[all...] |
| dnn_linear_combined.py | 34 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
[all...] |
| dynamic_rnn_estimator.py | 26 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
286 predictions_name = (prediction_key.PredictionKey.CLASSES
288 else prediction_key.PredictionKey.SCORES)
293 prediction_key.PredictionKey.PROBABILITIES: probabilities,
|
| state_saving_rnn_estimator_test.py | 30 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
583 prediction_key.PredictionKey.CLASSES,
584 prediction_key.PredictionKey.PROBABILITIES, ssre._get_state_name(0)
586 predictions = prediction_dict[prediction_key.PredictionKey.CLASSES]
587 probabilities = prediction_dict[prediction_key.PredictionKey.PROBABILITIES]
|
| debug_test.py | 494 prediction_key='classes'),
498 prediction_key='classes'),
501 metric_fn=_my_metric_op, prediction_key='probabilities')
523 prediction_key='bad_type')
795 prediction_key='scores'),
797 MetricSpec(metric_fn=_my_metric_op, prediction_key='scores')
809 # Tests the case where the prediction_key is not "scores".
818 prediction_key='bad_type')
|
| __init__.py | 327 from tensorflow.contrib.learn.python.learn.estimators.prediction_key import PredictionKey
|
| /external/tensorflow/tensorflow/examples/tutorials/monitors/ |
| iris_monitors.py | 43 prediction_key="classes"),
47 prediction_key="classes"),
51 prediction_key="classes")
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| eval_metrics.py | 23 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
30 INFERENCE_PROB_NAME = prediction_key.PredictionKey.PROBABILITIES
31 INFERENCE_PRED_NAME = prediction_key.PredictionKey.CLASSES
143 'r2': prediction_key.PredictionKey.SCORES,
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| sdca_estimator.py | 25 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
390 key = prediction_key.PredictionKey.CLASSES
405 key = prediction_key.PredictionKey.PROBABILITIES
517 key = prediction_key.PredictionKey.SCORES
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/ |
| kernel_estimators.py | 28 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
321 key = prediction_key.PredictionKey.CLASSES
336 key = prediction_key.PredictionKey.PROBABILITIES
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| export.py | 287 prediction_key=None,
316 if prediction_key is not None:
317 predictions = predictions[prediction_key]
|
| saved_model_export_utils.py | 42 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
130 scores = output_tensors.get(prediction_key.PredictionKey.SCORES)
132 scores = output_tensors.get(prediction_key.PredictionKey.PROBABILITIES)
137 classes = output_tensors.get(prediction_key.PredictionKey.CLASSES)
220 default_outputs = {prediction_key.PredictionKey.GENERIC: default_outputs}
[all...] |
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| estimators_test.py | 98 for prediction_key, prediction_value in estimator_predictions.items():
101 saved_prediction[prediction_key], axis=0))
|
| head.py | 115 for prediction_key, prediction_value in model_outputs.predictions.items():
116 metrics[prediction_key] = _identity_metric_single(prediction_key,
|
| /external/tensorflow/tensorflow/python/estimator/canned/ |
| head_test.py | [all...] |
| /external/tensorflow/tensorflow/examples/learn/ |
| random_forest_mnist.py | 73 prediction_key=eval_metrics.get_prediction_key(metric_name))
|
