text_plain_output_1.png image_output_1.png
from sklearn.linear_model import LinearRegression

model = LinearRegression()

model.fit(train_X, train_Y)

model.intercept_

model.coef_

train_predict = model.predict(train_X)