线性回归

 

import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets,linear_model,model_selection
#1、加载数据集
x,y=diabetes=datasets.load_diabetes(return_X_y=True)
print(x.shape)
print(y.shape)
x_train,x_test,y_train,y_test=model_selection.train_test_split(x,y,test_size=0.2)

#2、创建模型
model=linear_model.LinearRegression()
#3、训练模型
model.fit(x_train,y_train)
print(model.coef_)   #打印相关系数

#4、预测测试集
result=model.predict(x_test)
print(result)

#4、获取预测结果
score=model.score(x_test,y_test)
print(score)

plt.figure()
plt.plot(np.arange(0,len(y_test)),y_test,color='blue')
plt.plot(np.arange(0,len(y_test)),result,color='red')
plt.show()