Python基于numpy模块实现回归预测的方法

发布时间:2020-07-27 14:22:23 作者:小猪
来源:亿速云 阅读:256

这篇文章主要讲解了Python基于numpy模块实现回归预测的方法,内容清晰明了,对此有兴趣的小伙伴可以学习一下,相信大家阅读完之后会有帮助。

代码如下

import numpy as np
from matplotlib import pyplot as plt
# 用numpy生成数据t ,y
t = np.arange(1,10,1)
y = 0.9 * t + np.sin(t)
model = np.polyfit(t, y ,deg=1) # np.polyfit是numpy提供的加分分析方法,deg=1,指定模型为1阶的,返回值model为获得的模型
t2 = np.arange(-2,12,0.5) # 再生成一个间隔为0.5的序列
ypredict = np.polyval(model, t2) # 由np.polyval预测y值序列
plt.plot(t, y, "o", t2, ypredict, 'x')
plt.show()

上面的一段代码利用numpy生成数据序列,并实现了1阶回归,并画出预测效果图,图形如下:

Python基于numpy模块实现回归预测的方法

将代码改一下,实现2阶、3阶回归预测,只需要model = np.polyfit(t, y, deg =2)即可,同理3阶模型就把deg改为3即可。

2阶效果图和3阶效果图分别如下:

Python基于numpy模块实现回归预测的方法

Python基于numpy模块实现回归预测的方法

需要说明的是,并不是拟合的阶数越高,模型越好,本例使用2阶拟合效果比较好,如果使用3阶,会出现“过拟合”

看完上述内容,是不是对Python基于numpy模块实现回归预测的方法有进一步的了解,如果还想学习更多内容,欢迎关注亿速云行业资讯频道。

推荐阅读:
  1. pytorch-RNN进行回归曲线预测方式
  2. python用线性回归预测股票价格的实现代码

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

python numpy

上一篇:重定向详解

下一篇:unixbench详解

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》