diff --git a/examples/miscellaneous/plot_set_output.py b/examples/miscellaneous/plot_set_output.py index 12f2c822753e8..d36309e5b2745 100644 --- a/examples/miscellaneous/plot_set_output.py +++ b/examples/miscellaneous/plot_set_output.py @@ -109,3 +109,8 @@ log_reg = clf[-1] coef = pd.Series(log_reg.coef_.ravel(), index=log_reg.feature_names_in_) _ = coef.sort_values().plot.barh() + +# %% +# This resets `transform_output` to its default value to avoid impacting other +# examples when generating the scikit-learn documentation +set_config(transform_output="default")