diff --git a/pycse/sklearn/surface_response.py b/pycse/sklearn/surface_response.py index 88ca734..34158aa 100644 --- 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()), ]