线性回归模板
import matplotlib.pyplot as plt
import numpy as np
from sklearn import linear_model
clf=linear_model.LinearRegression()
x=np.array([2,3,5,7,6]).reshape(-1,1)
y=np.array([6,10,14.5,21,18.5])
plt.scatter(x,y,color='blue')
clf.fit(x,y)
b,a=clf.coef_,clf.intercept_ #y=bx+a
print(b,a)
x1=[[4]]
print(clf.predict(x))#预测
plt.plot(x,a+b*x,color='red')