diff options
author | 2018-09-27 19:20:59 -0700 | |
---|---|---|
committer | 2018-09-27 19:23:55 -0700 | |
commit | 70f071f7afb2deffddbd9937d7a76b1e1c0b2b75 (patch) | |
tree | 37f78a2ccfd09e15d1306fb8df43953efcf4a276 /tensorflow/contrib/boosted_trees | |
parent | 96f3428e33e18477661b8d8cf78f2db457c8881b (diff) |
Fix failing test.
PiperOrigin-RevId: 214875840
Diffstat (limited to 'tensorflow/contrib/boosted_trees')
-rw-r--r-- | tensorflow/contrib/boosted_trees/estimator_batch/dnn_tree_combined_estimator_test.py | 3 | ||||
-rw-r--r-- | tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py | 4 |
2 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/contrib/boosted_trees/estimator_batch/dnn_tree_combined_estimator_test.py b/tensorflow/contrib/boosted_trees/estimator_batch/dnn_tree_combined_estimator_test.py index 04baa329a0..6b6fe9663a 100644 --- a/tensorflow/contrib/boosted_trees/estimator_batch/dnn_tree_combined_estimator_test.py +++ b/tensorflow/contrib/boosted_trees/estimator_batch/dnn_tree_combined_estimator_test.py @@ -188,7 +188,8 @@ class CoreDNNBoostedTreeCombinedTest(test_util.TensorFlowTestCase): # Train for a few steps. est.train(input_fn=_train_input_fn, steps=1000) - # 10 steps for dnn, 3 for 1 tree of depth 3 + 1 after the tree finished + # 10 steps for dnn + 3 for 1 tree of depth 3 + 1 after the tree finished + # + 1 for resource variables. self._assert_checkpoint(est.model_dir, global_step=15) res = est.evaluate(input_fn=_eval_input_fn, steps=1) self.assertLess(0.5, res["auc"]) diff --git a/tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py b/tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py index c155128c0e..d7b14e00ba 100644 --- a/tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py +++ b/tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py @@ -238,8 +238,8 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase): output_leaf_index=False) classifier.fit(input_fn=_train_input_fn, steps=15) - # When no override of global steps, 5 steps were used. - self._assert_checkpoint(classifier.model_dir, global_step=5) + # When no override of global steps, 6 steps were used. + self._assert_checkpoint(classifier.model_dir, global_step=6) def testOverridesGlobalSteps(self): learner_config = learner_pb2.LearnerConfig() |