aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/boosted_trees
diff options
context:
space:
mode:
authorGravatar Revan Sopher <rsopher@google.com>2018-09-27 19:20:59 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-27 19:23:55 -0700
commit70f071f7afb2deffddbd9937d7a76b1e1c0b2b75 (patch)
tree37f78a2ccfd09e15d1306fb8df43953efcf4a276 /tensorflow/contrib/boosted_trees
parent96f3428e33e18477661b8d8cf78f2db457c8881b (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.py3
-rw-r--r--tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py4
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()