| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | head_test.py | 63 model_fn_ops): argument
64 test_case.assertAlmostEqual(expected_loss, model_fn_ops.loss.eval(), places=4)
66 test_case.assertIn(k, model_fn_ops.eval_metric_ops)
69 value_tensor, update_tensor = model_fn_ops.eval_metric_ops[key]
100 def _assert_output_alternatives(self, model_fn_ops): argument
104 k: v[0] for k, v in six.iteritems(model_fn_ops.output_alternatives)
120 model_fn_ops = head.create_model_fn_ops(
126 self._assert_output_alternatives(model_fn_ops)
130 _assert_metrics(self, loss, {"loss": loss}, model_fn_ops)
135 def _assert_output_alternatives(self, model_fn_ops): argument
[all …]
|
| D | model_fn_test.py | 65 self, model_fn_ops, estimator_spec): argument
67 for key, value in six.iteritems(model_fn_ops.eval_metric_ops):
70 self.assertEqual(model_fn_ops.predictions, estimator_spec.predictions)
71 self.assertEqual(model_fn_ops.loss, estimator_spec.loss)
72 self.assertEqual(model_fn_ops.train_op, estimator_spec.train_op)
75 self.assertAllEqual(model_fn_ops.training_chief_hooks,
77 self.assertAllEqual(model_fn_ops.training_hooks,
79 self.assertEqual(model_fn_ops.scaffold, estimator_spec.scaffold)
83 model_fn_ops = self.create_model_fn_ops(
86 estimator_spec = model_fn_ops.estimator_spec()
[all …]
|
| D | state_saving_rnn_estimator_test.py | 350 model_fn_ops = model_fn(features=features, labels=labels, mode=mode)
351 return model_fn_ops
356 model_fn_ops = self._getModelFnOpsForMode(model_fn_lib.ModeKeys.TRAIN)
357 self.assertIsNotNone(model_fn_ops.predictions)
358 self.assertIsNotNone(model_fn_ops.loss)
359 self.assertIsNotNone(model_fn_ops.train_op)
361 self.assertNotEqual(len(model_fn_ops.eval_metric_ops), 0)
364 model_fn_ops = self._getModelFnOpsForMode(model_fn_lib.ModeKeys.EVAL)
365 self.assertIsNotNone(model_fn_ops.predictions)
366 self.assertIsNotNone(model_fn_ops.loss)
[all …]
|
| D | estimator.py | 1041 model_fn_ops = self._get_train_ops(features, labels)
1042 ops.add_to_collection(ops.GraphKeys.LOSSES, model_fn_ops.loss)
1045 basic_session_run_hooks.NanTensorHook(model_fn_ops.loss),
1048 'loss': model_fn_ops.loss,
1054 scaffold = model_fn_ops.scaffold or monitored_session.Scaffold()
1071 for h in (all_hooks + model_fn_ops.training_hooks + chief_hooks +
1072 model_fn_ops.training_chief_hooks)
1087 hooks=all_hooks + model_fn_ops.training_hooks,
1088 chief_only_hooks=chief_hooks + model_fn_ops.training_chief_hooks,
1094 _, loss = mon_sess.run([model_fn_ops.train_op, model_fn_ops.loss])
[all …]
|
| D | dynamic_rnn_estimator_test.py | 225 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.TRAIN)
226 self.assertIsNotNone(model_fn_ops.predictions)
227 self.assertIsNotNone(model_fn_ops.loss)
228 self.assertIsNotNone(model_fn_ops.train_op)
230 self.assertNotEqual(len(model_fn_ops.eval_metric_ops), 0)
233 model_fn_ops = self._GetModelFnOpsForMode(model_fn_lib.ModeKeys.EVAL)
234 self.assertIsNotNone(model_fn_ops.predictions)
235 self.assertIsNotNone(model_fn_ops.loss)
236 self.assertIsNone(model_fn_ops.train_op)
238 self.assertNotEqual(len(model_fn_ops.eval_metric_ops), 0)
[all …]
|
| D | linear.py | 271 model_fn_ops = head.create_model_fn_ops(
278 return model_fn_ops._replace(
279 training_chief_hooks=(model_fn_ops.training_chief_hooks +
281 return model_fn_ops
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | model.py | 158 model_fn_ops = head.create_estimator_spec(
164 model_fn_ops = estimator_utils.estimator_spec_to_model_fn_ops(
165 model_fn_ops)
167 model_fn_ops = head.create_model_fn_ops(
175 model_fn_ops.predictions[gbdt_batch.LEAF_INDEX] = predictions_dict[
178 model_fn_ops.training_hooks.extend(training_hooks)
179 return model_fn_ops
193 return model_fn_ops
377 model_fn_ops = head.create_estimator_spec(
383 model_fn_ops = estimator_utils.estimator_spec_to_model_fn_ops(
[all …]
|
| D | dnn_tree_combined_estimator.py | 325 model_fn_ops = head.create_model_fn_ops(
332 return model_fn_ops
347 model_fn_ops.training_hooks.extend([
354 return model_fn_ops
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | saved_model_export_utils_test.py | 377 model_fn_ops = model_fn.ModelFnOps(
383 model_fn_ops, "head-1")
394 model_fn_ops = model_fn.ModelFnOps(
400 saved_model_export_utils.get_output_alternatives(model_fn_ops, "WRONG")
413 model_fn_ops = model_fn.ModelFnOps(
419 model_fn_ops)
434 model_fn_ops = model_fn.ModelFnOps(
440 saved_model_export_utils.get_output_alternatives(model_fn_ops)
448 model_fn_ops = model_fn.ModelFnOps(
454 model_fn_ops)
[all …]
|
| D | saved_model_export_utils.py | 192 def get_output_alternatives(model_fn_ops, default_output_alternative_key=None): argument
215 output_alternatives = model_fn_ops.output_alternatives
226 default_outputs = model_fn_ops.predictions
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | estimator_test.py | 40 def _assert_output_alternatives(self, model_fn_ops): argument
44 k: v[0] for k, v in six.iteritems(model_fn_ops.output_alternatives)
85 model_fn_ops = head.create_model_fn_ops(
91 self._assert_output_alternatives(model_fn_ops)
95 _assert_metrics(self, loss, {"loss": loss}, model_fn_ops)
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| D | random_forest.py | 501 model_fn_ops = []
505 model_fn_ops.append(
508 for mops in model_fn_ops:
516 for mops in model_fn_ops], axis=1)
519 for mops in model_fn_ops], axis=1)
525 [mops.loss for mops in model_fn_ops])) / len(model_fn_ops)
530 *[mops.train_op for mops in model_fn_ops])
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| D | contrib_estimator_predictor.py | 58 model_fn_ops = estimator._get_predict_ops(input_fn_ops.features)
76 model_fn_ops, output_alternative_key)
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_estimator.py | 168 model_fn_ops = head.create_model_fn_ops(
175 return model_fn_ops._replace(training_chief_hooks=(
176 model_fn_ops.training_chief_hooks + [update_weights_hook]))
177 return model_fn_ops
|