diff options
Diffstat (limited to 'tensorflow/python/estimator/canned/boosted_trees_test.py')
-rw-r--r-- | tensorflow/python/estimator/canned/boosted_trees_test.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/estimator/canned/boosted_trees_test.py b/tensorflow/python/estimator/canned/boosted_trees_test.py index f807641057..ec597e4686 100644 --- a/tensorflow/python/estimator/canned/boosted_trees_test.py +++ b/tensorflow/python/estimator/canned/boosted_trees_test.py @@ -1508,7 +1508,8 @@ class ModelFnTests(test_util.TensorFlowTestCase): l2=0.01, tree_complexity=0., min_node_weight=0., - center_bias=center_bias) + center_bias=center_bias, + pruning_mode='none') estimator_spec = boosted_trees._bt_model_fn( # pylint:disable=protected-access features=features, |