docs and changes in changes.rst

skrub-data · Dec 6, 2024 · 57f3e7f · 57f3e7f
1 parent 3c364be
commit 57f3e7f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git 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
@@ -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.