描述
在这里我们使用一个线性回归模型来拟合身高-体重的数据,也就是希望通过身高来预测标准体重。在已创建的数据集里已经给出了部分样本,所以一旦我们训练出一个模型,则可以对任意的身高数据都可以给出一个体重的预测值。这里需要分成3个部分。
第一部分是数据的创建;
第二部分是利用线性回归来拟合数据;
第三部分是利用已经训练好的模型去预测任意的体重并画出对任意体重的预测值,这里需要使用plot, 并且指定颜色为蓝色(color=“blue”)
from sklearn import datasets, linear_model # 引用 sklearn库,主要为了使用其中的线性回归模块
#创建数据集,把数据写入到numpy数组
import numpy as np # 引用numpy库,主要用来做科学计算
import matplotlib.pyplot as plt # 引用matplotlib库,主要用来画图
data = np.array([[152,51],[156,53],[160,54],[164,55],
[168,57],[172,60],[176,62],[180,65],
[184,69],[188,72]])
#打印出数组的大小
print(data.shape)
#TODO 1. 实例化一个线性回归的模型
#TODO 2. 在x,y上训练一个线性回归模型。 如果训练顺利,则regr会存储训练完成之后的结果模型
#TODO 3. 画出身高与体重之间的关系
#画出已训练好的线条
plt.plot(x,