diff --git a/stubs/sklearn/ensemble/_hist_gradient_boosting/gradient_boosting.pyi b/stubs/sklearn/ensemble/_hist_gradient_boosting/gradient_boosting.pyi index af87fe00..a5d0b9e9 100644 --- a/stubs/sklearn/ensemble/_hist_gradient_boosting/gradient_boosting.pyi +++ b/stubs/sklearn/ensemble/_hist_gradient_boosting/gradient_boosting.pyi @@ -54,6 +54,7 @@ class BaseHistGradientBoosting(BaseEstimator, ABC): max_depth, min_samples_leaf, l2_regularization, + max_features, max_bins, categorical_features, monotonic_cst, @@ -98,6 +99,7 @@ class HistGradientBoostingRegressor(RegressorMixin, BaseHistGradientBoosting): max_depth: None | int = None, min_samples_leaf: Int = 20, l2_regularization: Float = 0.0, + max_features: Float = 1.0, max_bins: Int = 255, categorical_features: None | MatrixLike | ArrayLike = None, monotonic_cst: None | Mapping | ArrayLike = None, @@ -144,6 +146,7 @@ class HistGradientBoostingClassifier(ClassifierMixin, BaseHistGradientBoosting): max_depth: None | int = None, min_samples_leaf: Int = 20, l2_regularization: Float = 0.0, + max_features: Float = 1.0, max_bins: Int = 255, categorical_features: None | MatrixLike | ArrayLike = None, monotonic_cst: None | Mapping | ArrayLike = None,