| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| metric_spec.py | 253 returns a prediction with key "prediction_key".
255 In order to compute the accuracy of the "prediction_key" prediction, we
260 prediction_key="prediction_key",
273 prediction_key="prediction_key",
290 prediction_key="prediction_key",
300 prediction_key=None,
315 prediction_key: The key for a tensor in the `predictions` dict (outpu
338 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/boosted_trees/estimator_batch/ |
| estimator_utils.py | 24 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
52 prediction_key.PredictionKey.SCORES: value.scores,
53 prediction_key.PredictionKey.CLASSES: value.classes
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| rnn_common.py | 29 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
36 # removed and replaced with values from `prediction_key.PredictionKey`. The key
38 # `prediction_key.PredictionKey.SCORES` for regression and
39 # `prediction_key.PredictionKey.CLASSES` for classification. The key
41 # `prediction_key.PredictionKey.PROBABILITIES`.
163 prediction_dict[prediction_key.PredictionKey.CLASSES], labels,
170 predictions=prediction_dict[prediction_key.PredictionKey.CLASSES],
253 containing single entry with key `prediction_key.PredictionKey.CLASSES` for
255 `prediction_key.PredictionKey.SCORE` for `problem_type`
259 `prediction_key.PredictionKey.PROBABILITIES`. Th
[all...] |
| debug.py | 57 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
220 key = prediction_key.PredictionKey.CLASSES
237 key = prediction_key.PredictionKey.PROBABILITIES
334 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,
493 pred_all_at_once[prediction_key.PredictionKey.CLASSES
[all...] |
| svm.py | 32 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
167 key = prediction_key.PredictionKey.CLASSES
184 key = prediction_key.PredictionKey.PROBABILITIES
|
| model_fn.py | 34 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
233 scores = output_tensors.get(prediction_key.PredictionKey.SCORES)
235 scores = output_tensors.get(prediction_key.PredictionKey.PROBABILITIES)
239 classes = output_tensors.get(prediction_key.PredictionKey.CLASSES)
|
| dnn.py | 38 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
457 key = prediction_key.PredictionKey.CLASSES
492 key = prediction_key.PredictionKey.PROBABILITIES
525 prediction_key=prediction_key.PredictionKey.PROBABILITIES,
690 custom_metrics[(key, prediction_key.PredictionKey.SCORES)] = metric
773 key = prediction_key.PredictionKey.SCORES
804 prediction_key=prediction_key.PredictionKey.SCORES,
[all...] |
| head.py | 34 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
777 key = prediction_key.PredictionKey.SCORES
[all...] |
| linear.py | 37 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
568 key = prediction_key.PredictionKey.CLASSES
599 key = prediction_key.PredictionKey.PROBABILITIES
631 prediction_key=prediction_key.PredictionKey.PROBABILITIES,
[all...] |
| dynamic_rnn_estimator.py | 31 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
291 predictions_name = (prediction_key.PredictionKey.CLASSES
293 else prediction_key.PredictionKey.SCORES)
298 prediction_key.PredictionKey.PROBABILITIES: probabilities,
|
| dnn_linear_combined.py | 39 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
[all...] |
| state_saving_rnn_estimator_test.py | 30 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
586 prediction_key.PredictionKey.CLASSES,
587 prediction_key.PredictionKey.PROBABILITIES, ssre._get_state_name(0)
589 predictions = prediction_dict[prediction_key.PredictionKey.CLASSES]
590 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 | 332 from tensorflow.contrib.learn.python.learn.estimators.prediction_key import PredictionKey
|
| 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...] |
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| eval_metrics.py | 23 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
31 INFERENCE_PROB_NAME = prediction_key.PredictionKey.PROBABILITIES
32 INFERENCE_PRED_NAME = prediction_key.PredictionKey.CLASSES
148 '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
401 key = prediction_key.PredictionKey.CLASSES
416 key = prediction_key.PredictionKey.PROBABILITIES
532 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 | 293 prediction_key=None,
323 if prediction_key is not None:
324 predictions = predictions[prediction_key]
|
| saved_model_export_utils.py | 46 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
136 scores = output_tensors.get(prediction_key.PredictionKey.SCORES)
138 scores = output_tensors.get(prediction_key.PredictionKey.PROBABILITIES)
143 classes = output_tensors.get(prediction_key.PredictionKey.CLASSES)
228 default_outputs = {prediction_key.PredictionKey.GENERIC: default_outputs}
[all...] |
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| head.py | 119 for prediction_key, prediction_value in model_outputs.predictions.items():
120 metrics[prediction_key] = _identity_metric_single(prediction_key,
|
| estimators_test.py | 115 for prediction_key, prediction_value in estimator_predictions.items():
118 saved_prediction[prediction_key], axis=0))
|
