Home
last modified time | relevance | path

Searched refs:PredictionKey (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead.py777 key = prediction_key.PredictionKey.SCORES
902 prediction_key.PredictionKey.LOGITS:
904 prediction_key.PredictionKey.LOGISTIC:
906 logits, name=prediction_key.PredictionKey.LOGISTIC),
907 prediction_key.PredictionKey.PROBABILITIES:
910 name=prediction_key.PredictionKey.PROBABILITIES),
911 prediction_key.PredictionKey.CLASSES:
915 name=prediction_key.PredictionKey.CLASSES)
922 classes = predictions[prediction_key.PredictionKey.CLASSES]
923 logistic = predictions[prediction_key.PredictionKey.LOGISTIC]
[all …]
Drnn_common.py163 prediction_dict[prediction_key.PredictionKey.CLASSES], labels,
170 predictions=prediction_dict[prediction_key.PredictionKey.CLASSES],
295 name=prediction_key.PredictionKey.PROBABILITIES)
297 prediction_key.PredictionKey.PROBABILITIES] = probabilities
301 predictions_name = (prediction_key.PredictionKey.CLASSES
303 else prediction_key.PredictionKey.SCORES)
Ddynamic_rnn_estimator_test.py193 {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]
495 pred_step_by_step[prediction_key.PredictionKey.CLASSES],
[all …]
Ddebug.py220 key = prediction_key.PredictionKey.CLASSES
237 key = prediction_key.PredictionKey.PROBABILITIES
334 key = prediction_key.PredictionKey.SCORES
Ddnn.py457 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,
Dmodel_fn.py233 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)
Ddnn_linear_combined.py778 key = prediction_key.PredictionKey.CLASSES
809 key = prediction_key.PredictionKey.PROBABILITIES
840 prediction_key=prediction_key.PredictionKey.PROBABILITIES,
1051 custom_metrics[(key, prediction_key.PredictionKey.SCORES)] = metric
1132 key = prediction_key.PredictionKey.SCORES
1162 prediction_key=prediction_key.PredictionKey.SCORES,
Dlinear.py568 key = prediction_key.PredictionKey.CLASSES
599 key = prediction_key.PredictionKey.PROBABILITIES
631 prediction_key=prediction_key.PredictionKey.PROBABILITIES,
839 key = prediction_key.PredictionKey.SCORES
870 prediction_key=prediction_key.PredictionKey.SCORES,
Dsvm.py167 key = prediction_key.PredictionKey.CLASSES
184 key = prediction_key.PredictionKey.PROBABILITIES
Dprediction_key.py28 class PredictionKey(object): class
Ddynamic_rnn_estimator.py291 predictions_name = (prediction_key.PredictionKey.CLASSES
293 else prediction_key.PredictionKey.SCORES)
298 prediction_key.PredictionKey.PROBABILITIES: probabilities,
Dstate_saving_rnn_estimator_test.py586 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]
Dhead_test.py1808 ("head1", prediction_key.PredictionKey.LOGITS),
1809 ("head1", prediction_key.PredictionKey.PROBABILITIES),
1810 ("head1", prediction_key.PredictionKey.CLASSES),
1811 ("head2", prediction_key.PredictionKey.LOGITS),
1812 ("head2", prediction_key.PredictionKey.PROBABILITIES),
1813 ("head2", prediction_key.PredictionKey.CLASSES),
1824 prediction_key.PredictionKey.PROBABILITIES,
1825 prediction_key.PredictionKey.CLASSES,
1828 prediction_key.PredictionKey.PROBABILITIES,
1829 prediction_key.PredictionKey.CLASSES,
D__init__.py332 from tensorflow.contrib.learn.python.learn.estimators.prediction_key import PredictionKey
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Deval_metrics.py31 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/boosted_trees/estimator_batch/
Destimator_utils.py52 prediction_key.PredictionKey.SCORES: value.scores,
53 prediction_key.PredictionKey.CLASSES: value.classes
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_estimator.py401 key = prediction_key.PredictionKey.CLASSES
416 key = prediction_key.PredictionKey.PROBABILITIES
532 key = prediction_key.PredictionKey.SCORES
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dsaved_model_export_utils.py136 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}
/external/tensorflow/tensorflow/contrib/kernel_methods/python/
Dkernel_estimators.py321 key = prediction_key.PredictionKey.CLASSES
336 key = prediction_key.PredictionKey.PROBABILITIES