Skip to content

Commit 0d546cf

Browse files
rstzcopybara-github
authored andcommitted
[YDF] [TF-DF] [PYDF] Fix max_depth parameter documentation
PiperOrigin-RevId: 608336146
1 parent 476debc commit 0d546cf

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

tensorflow_decision_forests/keras/wrappers_pre_generated.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,8 @@ class CartModel(core.CoreModel):
243243
model interpretation as well as hyper parameter tuning. This can take lots
244244
of space, sometimes accounting for half of the model size. Default: True.
245245
max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
246-
will be roots. Negative values are ignored. Default: 16.
246+
will be roots. `max_depth=-1` means that tree depth is not restricted by
247+
this parameter. Values <= -2 will be ignored. Default: 16.
247248
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
248249
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
249250
Default: None.
@@ -629,7 +630,8 @@ class DistributedGradientBoostedTreesModel(core.CoreModel):
629630
"max_unique_values_for_discretized_numerical" bins. This parameter will
630631
impact the model training. Default: False.
631632
max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
632-
will be roots. Negative values are ignored. Default: 6.
633+
will be roots. `max_depth=-1` means that tree depth is not restricted by
634+
this parameter. Values <= -2 will be ignored. Default: 6.
633635
max_unique_values_for_discretized_numerical: Maximum number of unique value
634636
of a numerical feature to allow its pre-discretization. In case of large
635637
datasets, discretized numerical features with a small number of unique
@@ -1078,7 +1080,8 @@ class GradientBoostedTreesModel(core.CoreModel):
10781080
a.k.a. MAE.
10791081
Default: "DEFAULT".
10801082
max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
1081-
will be roots. Negative values are ignored. Default: 6.
1083+
will be roots. `max_depth=-1` means that tree depth is not restricted by
1084+
this parameter. Values <= -2 will be ignored. Default: 6.
10821085
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
10831086
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
10841087
Default: None.
@@ -2136,7 +2139,8 @@ class RandomForestModel(core.CoreModel):
21362139
model interpretation as well as hyper parameter tuning. This can take lots
21372140
of space, sometimes accounting for half of the model size. Default: True.
21382141
max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
2139-
will be roots. Negative values are ignored. Default: 16.
2142+
will be roots. `max_depth=-1` means that tree depth is not restricted by
2143+
this parameter. Values <= -2 will be ignored. Default: 16.
21402144
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
21412145
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
21422146
Default: None.

0 commit comments

Comments
 (0)