From 57f3e7feb1b7d6bb5fcae63198f86112403c14f0 Mon Sep 17 00:00:00 2001 From: mrastgoo <mojdeh.rastgoo@gmail.com> Date: Fri, 6 Dec 2024 23:11:09 +0100 Subject: [PATCH] docs and changes in changes.rst --- CHANGES.rst | 5 +++++ skrub/_tabular_learner.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/CHANGES.rst b/CHANGES.rst index c2c43c21b..15568e0a1 100644 --- a/CHANGES.rst +++ b/CHANGES.rst @@ -12,6 +12,11 @@ Ongoing development Skrub is a very recent package. It is currently undergoing fast development and backward compatibility is not ensured. +Major changes +------------- +* :func:`tabular_learner` accepts the alias ``"regression"`` for the option ``"regressor"`` and ``"classification"`` for ``"classifier"``. + + Release 0.4.0 ============= diff --git a/skrub/_tabular_learner.py b/skrub/_tabular_learner.py index f6ca38ee7..fbec30c28 100644 --- a/skrub/_tabular_learner.py +++ b/skrub/_tabular_learner.py @@ -31,7 +31,7 @@ def tabular_learner(estimator, *, n_jobs=None): ``estimator``. Instead of an actual estimator, ``estimator`` can also be the special-cased strings - ``'regressor'`` or ``'classifier'`` to use a + ``'regressor'``, ``'regression'``, ``'classifier'``, ``'classification'`` to use a :obj:`~sklearn.ensemble.HistGradientBoostingRegressor` or a :obj:`~sklearn.ensemble.HistGradientBoostingClassifier` with default parameters.