diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-02 11:31:58 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-02 11:35:40 -0700 |
| commit | 61763fdd8e20bf1541dc12363d44318f81e06955 (patch) | |
| tree | 89501c20795f048ffe44f88d50c2eb86ba725f8c /tensorflow/contrib/boosted_trees | |
| parent | cf89b81401ea651dccf72ff322c1159618d0c8aa (diff) | |
Add a default value of None for the newly added parameter.
PiperOrigin-RevId: 207135538
Diffstat (limited to 'tensorflow/contrib/boosted_trees')
| -rw-r--r-- | tensorflow/contrib/boosted_trees/estimator_batch/trainer_hooks.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/boosted_trees/estimator_batch/trainer_hooks.py b/tensorflow/contrib/boosted_trees/estimator_batch/trainer_hooks.py index cb9f020b88..f137ada355 100644 --- a/tensorflow/contrib/boosted_trees/estimator_batch/trainer_hooks.py +++ b/tensorflow/contrib/boosted_trees/estimator_batch/trainer_hooks.py @@ -152,7 +152,7 @@ class StopAfterNTrees(session_run_hook.SessionRunHook): """Stop training after building N full trees.""" def __init__(self, n, num_attempted_trees_tensor, num_finalized_trees_tensor, - override_global_step_value): + override_global_step_value=None): self._num_trees = n # num_attempted_trees_tensor and num_finalized_trees_tensor are both # tensors. |