sklearn Polynomial Regression

poly.py

class PolynomialRegression(BaseEstimator):

    def __init__(self, degree=2):
        self.features = PolynomialFeatures(degree=degree)
        self.model = LinearRegression()

    def get_params(self, deep=True):
        return {'degree': self.features.degree}

    def set_params(self, **params):
        self.features.degree = params['degree']
        return self

    def fit(self, x, y):
        x_poly = self.features.fit_transform(np.array(x).reshape(-1, 1))
        return self.model.fit(x_poly, y)

    def predict(self, x):
        x_poly = self.features.fit_transform(np.array(x).reshape(-1, 1))
        return self.model.predict(x_poly)