diff options
author | 2018-09-28 10:05:17 -0700 | |
---|---|---|
committer | 2018-09-28 10:09:04 -0700 | |
commit | b47f0b1b0ac8047d53a824f4ca82a12387a16e4d (patch) | |
tree | 71ec3a8110e385d917534a3ff970dc189ae62b94 /tensorflow/contrib/boosted_trees/estimator_batch | |
parent | 7052b44b032a35edb10893ce08993a54e2a76e1d (diff) |
Updating the V2 variables API for boosted_trees.
PiperOrigin-RevId: 214952666
Diffstat (limited to 'tensorflow/contrib/boosted_trees/estimator_batch')
-rw-r--r-- | tensorflow/contrib/boosted_trees/estimator_batch/dnn_tree_combined_estimator_test.py | 2 |
1 files changed, 1 insertions, 1 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 6b6fe9663a..83a8dee632 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 @@ -190,7 +190,7 @@ class CoreDNNBoostedTreeCombinedTest(test_util.TensorFlowTestCase): 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. - self._assert_checkpoint(est.model_dir, global_step=15) + 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"]) est.predict(input_fn=_eval_input_fn) |