fix output of scaler so we get user labels in summary

jkitchin · Jul 10, 2024 · 618d71f · 618d71f
1 parent 7176cbe
commit 618d71f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pycse/sklearn/surface_response.py b/pycse/sklearn/surface_response.py
@@ -60,9 +60,10 @@ def __init__(
 
         if model is None:
             self.default = True
+            scaler = MinMaxScaler(feature_range=(-1, 1)).set_output(transform="pandas")
             super().__init__(
                 steps=[
-                    ("minmax", MinMaxScaler(feature_range=(-1, 1))),
+                    ("minmax", scaler),
                     ("poly", PolynomialFeatures(2)),
                     ("surface response", LinearRegressionUQ()),
                 ]