Add types for HistGradientBoosting max_features argument (#323)

microsoft · Oct 29, 2024 · 6702500 · 6702500
1 parent df107ae
commit 6702500
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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,