text_plain_output_1.png
from sklearn.linear_model import *
from sklearn.metrics import *
regr = LinearRegression()
regr.fit(X_train, y_train)
pred = regr.predict(X_test)
mean_squared_error(pred, y_test)