# %load q03_linear_regression/build.py
from greyatomlib.linear_regression.q01_load_data.build import load_data
from greyatomlib.linear_regression.q02_data_splitter.build import data_splitter
from sklearn.linear_model import LinearRegression

dataframe = load_data('data/house_prices_multivariate.csv')
X, y = data_splitter(dataframe)
def linear_regression(X,y):
    lr = LinearRegression()
    model = lr.fit(X,y)
    return model
    #lr.predict(X)

linear_regression(X,y)
