Skip to content

Commit 10fb357

Browse files
authored
Merge pull request #541 from KhiopsML/534-add-missing-attributes-to-khiopsregressor
534 add missing attributes to khiopsregressor
2 parents 019b089 + 851d504 commit 10fb357

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

khiops/sklearn/estimators.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2064,6 +2064,18 @@ class KhiopsRegressor(RegressorMixin, KhiopsPredictor):
20642064
Maximum number of features to construct automatically. See
20652065
:doc:`/multi_table_primer` for more details on the multi-table-specific
20662066
features.
2067+
n_trees : int, default 10
2068+
Maximum number of decision tree features to construct. The constructed trees
2069+
combine other features, either native or constructed. These features usually
2070+
improve the regressor's performance at the cost of interpretability of the
2071+
model.
2072+
n_text_features : int, default 10000
2073+
Maximum number of text features to construct.
2074+
type_text_features : str, default "words"
2075+
Type of the text features to construct. Can be either one of:
2076+
- "words": sequences of non-space characters
2077+
- "ngrams": sequences of bytes
2078+
- "tokens": user-defined
20672079
n_selected_features : int, default 0
20682080
Maximum number of features to be selected in the SNB predictor. If equal to
20692081
0 it selects all the features kept in the training.
@@ -2160,8 +2172,6 @@ def fit(self, X, y=None, **kwargs):
21602172
self : `KhiopsRegressor`
21612173
The calling estimator instance.
21622174
"""
2163-
if self.n_trees > 0:
2164-
warnings.warn("Khiops does not support n_trees > 0 for regression models.")
21652175
kwargs["categorical_target"] = False
21662176
return super().fit(X, y=y, **kwargs)
21672177

0 commit comments

Comments
 (0)