From 4be7104ed1196802520d6ec3cbd89808e81a9e11 Mon Sep 17 00:00:00 2001 From: "Thomas J. Fan" Date: Wed, 20 Dec 2023 15:05:51 -0500 Subject: [PATCH] DOC Fixes pipeline docstring for pandas output --- sklearn/pipeline.py | 1 + 1 file changed, 1 insertion(+) 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