Home
last modified time | relevance | path

Searched refs:prediction_key (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drnn_common.py29 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
163 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)
Dhead.py34 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
777 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]
[all …]
Ddynamic_rnn_estimator_test.py31 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 …]
Ddnn.py38 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,
Ddebug.py57 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
Dlinear.py37 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,
839 key = prediction_key.PredictionKey.SCORES
870 prediction_key=prediction_key.PredictionKey.SCORES,
Ddnn_linear_combined.py39 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
778 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,
Dmodel_fn.py34 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)
Dsvm.py32 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
167 key = prediction_key.PredictionKey.CLASSES
184 key = prediction_key.PredictionKey.PROBABILITIES
Ddynamic_rnn_estimator.py31 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,
Dstate_saving_rnn_estimator_test.py30 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]
Ddebug_test.py494 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')
818 prediction_key='bad_type')
Dhead_test.py30 from tensorflow.contrib.learn.python.learn.estimators import prediction_key
1808 ("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,
[all …]
Ddnn_linear_combined_test.py828 prediction_key='classes'),
832 prediction_key='classes'),
835 metric_fn=_my_metric_op, prediction_key='probabilities')
875 prediction_key='bad_type')
1510 prediction_key='scores'),
1513 metric_fn=_my_metric_op, prediction_key='scores')
1534 prediction_key='bad_type')
Ddnn_test.py837 prediction_key='classes'),
841 prediction_key='classes'),
844 metric_fn=_my_metric_op, prediction_key='probabilities')
866 prediction_key='bad_type')
1408 prediction_key='scores'),
1411 metric_fn=_my_metric_op, prediction_key='scores')
1432 prediction_key='bad_type')
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dmetric_spec_test.py61 metric_fn=fn, prediction_key="p1", label_key="l1", weight_key="f2")
212 spec = MetricSpec(metric_fn=fn, prediction_key="p1", label_key="l1")
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")
307 metric_fn=_fn, prediction_key="p1", label_key="l1", weight_key="f2")
319 prediction_key="my_prediction",
334 prediction_key="my_prediction",
357 prediction_key="p1",
365 prediction_key="p1", label_key="l1", weight_key="f2")
Dmetric_spec.py300 prediction_key=None, argument
333 self._prediction_key = prediction_key
338 def prediction_key(self): member in MetricSpec
364 'prediction_key=%s, ' % self.prediction_key +
430 prediction = _get_dict('prediction', predictions, self.prediction_key)
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_utils.py24 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/tensor_forest/client/
Deval_metrics.py23 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/kernel_methods/python/
Dkernel_estimators.py28 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/linear_optimizer/python/
Dsdca_estimator.py25 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/learn/python/learn/utils/
Dsaved_model_export_utils.py46 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}
Dexport.py293 prediction_key=None, argument
323 if prediction_key is not None:
324 predictions = predictions[prediction_key]
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dhead.py119 for prediction_key, prediction_value in model_outputs.predictions.items():
120 metrics[prediction_key] = _identity_metric_single(prediction_key,
Destimators_test.py115 for prediction_key, prediction_value in estimator_predictions.items():
118 saved_prediction[prediction_key], axis=0))

12