1a2462f18707e04f294224053473668820111cf5,PyPi/approximators/regressor.py,Regressor,predict,#Regressor#Any#,54

Before Change

        // Returns
            The prediction of the model.
        
        if x.ndim == 1:
            x = np.expand_dims(x, axis=0)
        elif x.ndim > 2:
            raise ValueError("Training set dimension not suitable for the "
                             "regressor.")

        if self.features:
            x = self.features.transform(x)

        if self.input_scaled:

After Change

            assert x[0].ndim == 2 and x[1].ndim == 2
            assert x[0].shape[0] == x[1].shape[0]

            x = np.concatenate((x[0], x[1]), axis=1)

        if isinstance(x, list):
            if self.features:
                x[0] = self.features.transform(x[0])

            if self.input_scaled:
                self.pre_x = preprocessing.StandardScaler()
                x[0] = self.pre_x.transform(x[0])
        else:
            if self.features:
                x = self.features.transform(x)

            if self.input_scaled:
                self.pre_x = preprocessing.StandardScaler()
                x = self.pre_x.transform(x)

        y = self.model.predict(x)

        return self.pre_y.inverse_transform(y) if self.output_scaled else y

Instances