Answers for "basic regression in python"

python linear regression

from sklearn.linear_model import LinearRegression
#x, y = np.array([0,1,2,3,4,5])

model = LinearRegression().fit(x.reshape(-1,1), y.reshape(-1,1))
r_sq = model.score(x.reshape(-1,1), y.reshape(-1,1))
q = model.intercept_
m = model.coef_

y_fit = np.array([i*m[0] for i in x]+q[0])

real python linear regression

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)