application_vnd.jupyter.stderr_output_1.png
from sklearn.linear_model import LinearRegression

from sklearn.metrics import mean_absolute_error, mean_squared_error

model = LinearRegression()

model.fit(train_X, train_Y)

model.intercept_

model.coef_

train_predict = model.predict(train_X)

test_predict = model.predict(test_X)

print(mean_squared_error(train_Y, train_predict))
print(mean_squared_error(test_Y, test_predict))