| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dynamic_rnn_estimator.py | 263 problem_type, argument
292 if problem_type == constants.ProblemType.CLASSIFICATION
355 problem_type, argument
380 return {'dynamic_rnn_output': (problem_type, prediction_dict_no_state)}
388 problem_type, argument
462 if problem_type not in (constants.ProblemType.CLASSIFICATION,
467 format(problem_type))
474 if (problem_type != constants.ProblemType.CLASSIFICATION
478 ' ProblemType.CLASSIFICATION; got {}.'.format(problem_type))
504 rnn_activations, target_column, problem_type, predict_probabilities)
[all …]
|
| D | state_saving_rnn_estimator.py | 388 problem_type, argument
453 if problem_type not in (constants.ProblemType.CLASSIFICATION,
458 format(problem_type))
459 if (problem_type != constants.ProblemType.CLASSIFICATION and
463 ' ProblemType.CLASSIFICATION; got {}.'.format(problem_type))
503 rnn_activations, target_column, problem_type, predict_probabilities)
511 problem_type, rnn_common.PredictionType.MULTIPLE_VALUE,
544 problem_type, argument
623 if problem_type == constants.ProblemType.LINEAR_REGRESSION:
626 elif problem_type == constants.ProblemType.CLASSIFICATION:
[all …]
|
| D | model_fn.py | 252 def _export_output(problem_type, predictions): # pylint: disable=missing-docstring argument
253 if problem_type == constants.ProblemType.LINEAR_REGRESSION:
256 if (problem_type == constants.ProblemType.CLASSIFICATION or
257 problem_type == constants.ProblemType.LOGISTIC_REGRESSION):
261 if problem_type == constants.ProblemType.UNSPECIFIED:
264 raise ValueError('Unknown problem_type=%s' % problem_type)
|
| D | rnn_common.py | 140 def get_eval_metric_ops(problem_type, prediction_type, sequence_length, argument
159 if problem_type == constants.ProblemType.CLASSIFICATION:
172 elif problem_type == constants.ProblemType.LINEAR_REGRESSION:
248 def multi_value_predictions(activations, target_column, problem_type, argument
302 if problem_type == constants.ProblemType.CLASSIFICATION
|
| D | dynamic_rnn_estimator_test.py | 255 problem_type=constants.ProblemType.CLASSIFICATION,
286 problem_type=constants.ProblemType.CLASSIFICATION,
393 problem_type=constants.ProblemType.CLASSIFICATION,
449 problem_type=constants.ProblemType.CLASSIFICATION,
546 problem_type=constants.ProblemType.LINEAR_REGRESSION,
607 problem_type=constants.ProblemType.CLASSIFICATION,
686 problem_type=constants.ProblemType.LINEAR_REGRESSION,
743 problem_type=constants.ProblemType.CLASSIFICATION,
|
| D | head.py | 509 self, problem_type, logits_dimension, label_name=None, argument
511 if problem_type is None:
512 raise ValueError("Invalid problem_type %s." % problem_type)
515 self._problem_type = problem_type
718 problem_type=constants.ProblemType.LINEAR_REGRESSION,
841 problem_type=constants.ProblemType.LOGISTIC_REGRESSION,
1035 problem_type=constants.ProblemType.CLASSIFICATION,
1243 problem_type=constants.ProblemType.LOGISTIC_REGRESSION,
1337 problem_type=constants.ProblemType.CLASSIFICATION,
2069 def _classification_output_alternatives(head_name, problem_type, argument
[all …]
|
| D | state_saving_rnn_estimator_test.py | 346 problem_type=constants.ProblemType.CLASSIFICATION,
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | saved_model_export_utils.py | 87 problem_type): argument
114 if _is_classification_problem(problem_type, input_tensors, output_tensors):
126 elif _is_regression_problem(problem_type, input_tensors, output_tensors):
150 def _is_classification_problem(problem_type, input_tensors, output_tensors): argument
153 return ((problem_type == constants.ProblemType.CLASSIFICATION or
154 problem_type == constants.ProblemType.LOGISTIC_REGRESSION) and
160 def _is_regression_problem(problem_type, input_tensors, output_tensors): argument
161 return (problem_type == constants.ProblemType.LINEAR_REGRESSION and
264 problem_type)
266 for output_key, (problem_type,
|
| D | saved_model_export_utils_test.py | 84 problem_type = constants.ProblemType.LINEAR_REGRESSION
87 input_tensors, output_tensors, problem_type))
115 problem_type = constants.ProblemType.CLASSIFICATION
118 input_tensors, output_tensors, problem_type))
155 problem_type = constants.ProblemType.CLASSIFICATION
158 input_tensors, output_tensors, problem_type))
201 problem_type = constants.ProblemType.CLASSIFICATION
204 input_tensors, output_tensors, problem_type))
244 problem_type = constants.ProblemType.CLASSIFICATION
247 input_tensors, output_tensors, problem_type))
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | target_column.py | 152 problem_type): argument
162 self._problem_type = problem_type
193 def problem_type(self): member in _TargetColumn
272 problem_type=ProblemType.LINEAR_REGRESSION)
302 problem_type=ProblemType.CLASSIFICATION)
|