Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 12 additions & 2 deletions khiops/sklearn/estimators.py
Original file line number Diff line number Diff line change
Expand Up @@ -2064,6 +2064,18 @@ class KhiopsRegressor(RegressorMixin, KhiopsPredictor):
Maximum number of features to construct automatically. See
:doc:`/multi_table_primer` for more details on the multi-table-specific
features.
n_trees : int, default 10
Maximum number of decision tree features to construct. The constructed trees
combine other features, either native or constructed. These features usually
improve the regressor's performance at the cost of interpretability of the
model.
n_text_features : int, default 10000
Maximum number of text features to construct.
type_text_features : str, default "words"
Type of the text features to construct. Can be either one of:
- "words": sequences of non-space characters
- "ngrams": sequences of bytes
- "tokens": user-defined
n_selected_features : int, default 0
Maximum number of features to be selected in the SNB predictor. If equal to
0 it selects all the features kept in the training.
Expand Down Expand Up @@ -2160,8 +2172,6 @@ def fit(self, X, y=None, **kwargs):
self : `KhiopsRegressor`
The calling estimator instance.
"""
if self.n_trees > 0:
warnings.warn("Khiops does not support n_trees > 0 for regression models.")
kwargs["categorical_target"] = False
return super().fit(X, y=y, **kwargs)

Expand Down
Loading