Skip to content

Commit 3debd21

Browse files
Update legate-boost.rst
1 parent 81b587e commit 3debd21

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

docs/source/legate-boost.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ The following code creates a ``legate-boost`` regression model using
182182
``LBRegressor``, which trains a gradient boosting model optimized for
183183
multi-GPU and multi-node environments. The model is configured to use
184184
100 boosting rounds (n_estimators=100), with each round adding a
185-
decision tree (lb.models.Tree) limited to a maximum depth of 5. The loss
185+
decision tree (lb.models.Tree) limited to a maximum depth of 8. The loss
186186
function is set to squared_error, suitable for regression tasks as it
187187
minimizes mean squared error. A learning_rate of 0.1 controls how much
188188
each tree contributes to the final prediction, balancing speed and
@@ -461,7 +461,7 @@ individually.
461461
462462
model = lb.LBClassifier(
463463
base_models=(
464-
lb.models.Tree(max_depth=8),
464+
lb.models.Tree(max_depth=5),
465465
lb.models.NN(max_iter=10, hidden_layer_sizes=(10,10), verbose=True)
466466
)
467467
).fit(x_train, y_train)

0 commit comments

Comments
 (0)