diff options
author | Revan Sopher <rsopher@google.com> | 2018-09-28 18:27:37 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 18:32:08 -0700 |
commit | abd5c32c0fa6451e73b491affdd86d852a74177f (patch) | |
tree | 94f0f51fad583ff14a822fc9ebcd8aae329f1854 /tensorflow/contrib/boosted_trees | |
parent | b34ddf043324e52ee0acdfe62cb18beab7fed08e (diff) |
Automated rollback of commit 70f071f7afb2deffddbd9937d7a76b1e1c0b2b75
PiperOrigin-RevId: 215026418
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, 3 insertions, 4 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 83a8dee632..839eedd3a8 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,8 +188,7 @@ 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 - # + 1 for resource variables. + # 10 steps for dnn, 3 for 1 tree of depth 3 + 1 after the tree finished self._assert_checkpoint(est.model_dir, global_step=14) 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 d7b14e00ba..c155128c0e 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, 6 steps were used. - self._assert_checkpoint(classifier.model_dir, global_step=6) + # When no override of global steps, 5 steps were used. + self._assert_checkpoint(classifier.model_dir, global_step=5) def testOverridesGlobalSteps(self): learner_config = learner_pb2.LearnerConfig() |