class Polinomial_Regression(object):
def __init__(self):
self.lin_reg = None
self.poli_reg = None
self.create_regressor()
def create_regressor(self):
self.lin_reg = LinearRegression()
def fit_linear(self, x_train, y_train):
self.lin_reg.fit(x_train, y_train)
def create_polinominal_reg(self):
self.poli_reg = PolynomialFeatures(degree=2)
# the contant b0 is added bythe fit_transform method;
def fit_transform_poly(self, x_train, y_train):
x_poly = self.poli_reg.fit_tranform(x_train)
linear_regression = LinearRegression()
linear_regression.fit(x_poly, y_train)
"""
