diff --git a/feature_engine/creation/decision_tree_features.py b/feature_engine/creation/decision_tree_features.py index e0f844005..b1b3a2c92 100644 --- a/feature_engine/creation/decision_tree_features.py +++ b/feature_engine/creation/decision_tree_features.py @@ -135,13 +135,9 @@ class DecisionTreeFeatures(TransformerMixin, BaseEstimator, GetFeatureNamesOutMi the random_state to an integer. n_jobs: int, default=None - The number of jobs to run in parallel when training the decision trees - across feature combinations. Trees are fit using threads rather than - processes, since fitting a decision tree releases the GIL for the bulk - of its computation, which avoids the overhead of copying the entire - dataframe to separate worker processes. `None` means 1, i.e. sequential - training (this transformer's original behaviour); `-1` means using all - available processors. + The number of jobs to run in parallel. `fit` is parallelized over the feature + combinations, training one decision tree per combination. `None` means 1 + unless in a `joblib.parallel_backend` context. `-1` means using all processors. {missing_values}