Skip to content

Commit

Permalink
[YDF] [TF-DF] [PYDF] Fix max_depth parameter documentation
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 608336146
  • Loading branch information
rstz authored and copybara-github committed Feb 19, 2024
1 parent 476debc commit 0d546cf
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions tensorflow_decision_forests/keras/wrappers_pre_generated.py
Original file line number Diff line number Diff line change
Expand Up @@ -243,7 +243,8 @@ class CartModel(core.CoreModel):
model interpretation as well as hyper parameter tuning. This can take lots
of space, sometimes accounting for half of the model size. Default: True.
max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
will be roots. Negative values are ignored. Default: 16.
will be roots. `max_depth=-1` means that tree depth is not restricted by
this parameter. Values <= -2 will be ignored. Default: 16.
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
Default: None.
Expand Down Expand Up @@ -629,7 +630,8 @@ class DistributedGradientBoostedTreesModel(core.CoreModel):
"max_unique_values_for_discretized_numerical" bins. This parameter will
impact the model training. Default: False.
max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
will be roots. Negative values are ignored. Default: 6.
will be roots. `max_depth=-1` means that tree depth is not restricted by
this parameter. Values <= -2 will be ignored. Default: 6.
max_unique_values_for_discretized_numerical: Maximum number of unique value
of a numerical feature to allow its pre-discretization. In case of large
datasets, discretized numerical features with a small number of unique
Expand Down Expand Up @@ -1078,7 +1080,8 @@ class GradientBoostedTreesModel(core.CoreModel):
a.k.a. MAE.
Default: "DEFAULT".
max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
will be roots. Negative values are ignored. Default: 6.
will be roots. `max_depth=-1` means that tree depth is not restricted by
this parameter. Values <= -2 will be ignored. Default: 6.
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
Default: None.
Expand Down Expand Up @@ -2136,7 +2139,8 @@ class RandomForestModel(core.CoreModel):
model interpretation as well as hyper parameter tuning. This can take lots
of space, sometimes accounting for half of the model size. Default: True.
max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
will be roots. Negative values are ignored. Default: 16.
will be roots. `max_depth=-1` means that tree depth is not restricted by
this parameter. Values <= -2 will be ignored. Default: 16.
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
Default: None.
Expand Down

0 comments on commit 0d546cf

Please sign in to comment.