diff --git a/sklearn/pipeline.py b/sklearn/pipeline.py index 0bd81a3c57918..4df21618be4ee 100644 --- a/sklearn/pipeline.py +++ b/sklearn/pipeline.py @@ -182,6 +182,7 @@ def set_output(self, *, transform=None): Configure output of `transform` and `fit_transform`. - `"default"`: Default output format of a transformer + - `"pandas"`: DataFrame output - `"polars"`: Polars output - `None`: Transform configuration is unchanged