python中常用的三种线性回归

from sklearn.linear_model import LinearRegression
import numpy as np
import matplotlib.pyplot as plt
X = [[1],[4],[3]] #输入X
y = [3,5,3]     #输入y
lr = LinearRegression().fit(X,y)
z = np.linspace(0,5,20) 
plt.scatter(X,y,s=80)
plt.plot(z, lr.predict(z.reshape(-1,1)),c='k')#reshape将z改为任意行，一列
plt.title('Straight Line')
plt.show()
print('\n\n直线方程为：')
print('==============')
print('y = {:.3f}'.format(lr.coef_[0]),'x','+{:.3f}'.format(lr.intercept_))
print('==============')