线性回归

这篇博客介绍了如何运用线性回归模型对身高和体重数据进行拟合。首先,通过numpy创建数据集,然后使用sklearn的linear_model模块实例化线性回归模型,并在数据上训练模型。最后,通过matplotlib将身高与预测体重的关系以图表形式展示出来,预测了身高为163厘米的人的标准体重。
摘要由CSDN通过智能技术生成
描述

在这里我们使用一个线性回归模型来拟合身高-体重的数据,也就是希望通过身高来预测标准体重。在已创建的数据集里已经给出了部分样本,所以一旦我们训练出一个模型,则可以对任意的身高数据都可以给出一个体重的预测值。这里需要分成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,

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值