@@ -243,7 +243,8 @@ class CartModel(core.CoreModel):
243
243
model interpretation as well as hyper parameter tuning. This can take lots
244
244
of space, sometimes accounting for half of the model size. Default: True.
245
245
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.
247
248
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
248
249
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
249
250
Default: None.
@@ -629,7 +630,8 @@ class DistributedGradientBoostedTreesModel(core.CoreModel):
629
630
"max_unique_values_for_discretized_numerical" bins. This parameter will
630
631
impact the model training. Default: False.
631
632
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.
633
635
max_unique_values_for_discretized_numerical: Maximum number of unique value
634
636
of a numerical feature to allow its pre-discretization. In case of large
635
637
datasets, discretized numerical features with a small number of unique
@@ -1078,7 +1080,8 @@ class GradientBoostedTreesModel(core.CoreModel):
1078
1080
a.k.a. MAE.
1079
1081
Default: "DEFAULT".
1080
1082
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.
1082
1085
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
1083
1086
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
1084
1087
Default: None.
@@ -2136,7 +2139,8 @@ class RandomForestModel(core.CoreModel):
2136
2139
model interpretation as well as hyper parameter tuning. This can take lots
2137
2140
of space, sometimes accounting for half of the model size. Default: True.
2138
2141
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.
2140
2144
max_num_nodes: Maximum number of nodes in the tree. Set to -1 to disable
2141
2145
this limit. Only available for `growing_strategy=BEST_FIRST_GLOBAL`.
2142
2146
Default: None.
0 commit comments