Python实现批量梯度下降法(BGD)拟合曲线
作者:古明地板砖 时间:2023-03-31 07:18:03
1. 导入库
import numpy as np #矩阵运算
import matplotlib.pyplot as plt #可视化
import random #产生数据扰动
2. 产生数据
拟合曲线 y = 2 × x2 + x + 1
X_m = np.mat([[i**2, i, 1] for i in range(-10,10)]) #矩阵类型,用于运算
y_m = np.mat([[2*x[0,0]+x[0,1]+1+random.normalvariate(0,1)] for x in X_m]) #矩阵类型,用于运算
X_a = np.asarray(X_m[:,1].T)[0] #array类型,用于可视化
y_a = np.asarray(y_m.T)[0] #array类型,用于可视化
plt.scatter(X_a, y_a) #显示数据
plt.show()
3. BGD
def BGD(X,y,w0,step,e): #批量梯度下降法
n=0
while n<=10000:
w1 = w0-step*X.T.dot(X.dot(w0)-y)/X.shape[0]
dw = w1-w0;
if dw.dot(dw.T)[0,0] <= e**2:
return w1
n += 1
w0 = w1
return w1
4. 计算
w_m = BGD(X_m,y_m,np.mat([[5],[3],[2]]),1e-4,1e-20) #可自行调参
w_a = np.asarray(w_m.T)[0]
print(w_a)
array([1.99458492, 0.91587829, 1.48498921])
5. 评价( R 2)
y_mean = y_a.mean()
y_pre = np.array([w_a[0]*x[0,0]+w_a[1]*x[0,1]+w_a[2] for x in X_m])
SSR = ((y_pre-y_mean)**2).sum()
SST = ((y_a-y_mean)**2).sum()
R2 = SSR/SST
print(R2)
0.9845542903194531
我们可以认为拟合效果不错。如果 R 2 R^{2} R2的值接近0,可能需要重新调参。
6. 结果展示
X = np.linspace(-10,10,50)
y = np.array([w_a[0]*x**2+w_a[1]*x+w_a[2] for x in X])
plt.scatter(X_a,y_a)
plt.plot(X,y)
plt.show()
来源:https://blog.csdn.net/Twilightzsj/article/details/115326848
标签:Python,批量,梯度下降
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
一个奇怪的CSS现象
2010-02-10 12:28:00
![](https://img.aspxhome.com/file/UploadPic/20102/10/y200921122655-79s.jpg)
python中把元组转换为namedtuple方法
2021-12-04 22:13:04
Python一行代码对话ChatGPT实现详解
2022-04-24 04:07:26
![](https://img.aspxhome.com/file/2023/7/85367_0s.jpg)
mssql2005字符串连接方法 避免无效的连接错误
2024-01-14 20:45:05
基于pako.js实现gzip的压缩和解压功能示例
2024-04-22 22:14:41
![](https://img.aspxhome.com/file/2023/4/136064_0s.jpg)
深入了解和应用Python 装饰器 @decorator
2024-01-02 09:20:05
![](https://img.aspxhome.com/file/2023/2/72152_0s.jpg)
python装饰器原理与用法深入详解
2021-01-13 09:17:19
Python 功能和特点(新手必学)
2022-02-23 04:11:25
Python import与from import使用及区别介绍
2022-05-17 12:09:19
PyCharm2021最新激活码+激活码补丁(亲测最新版PyCharm2021.3激活成功)
2022-09-18 05:22:03
![](https://img.aspxhome.com/file/2023/2/80772_0s.png)
如何判断用户在某一页面逗留了多长时间?
2010-01-18 20:48:00
JavaScript实现复制内容到粘贴板代码
2024-04-25 13:14:01
如何基于python操作json文件获取内容
2023-03-30 15:48:04
Python threading中lock的使用详解
2023-01-16 08:32:26
javascript new fun的执行过程
2010-08-05 21:23:00
![](https://img.aspxhome.com/file/UploadPic/20108/5/20100805150736846-93s.jpg)
Python @property装饰器原理解析
2022-02-19 20:28:05
MySQL索引的一些常见面试题大全(2022年)
2024-01-13 00:17:30
![](https://img.aspxhome.com/file/2023/8/111378_0s.jpg)
go语言区块链实战实现简单的区块与区块链
2024-05-10 13:57:33
![](https://img.aspxhome.com/file/2023/4/127484_0s.png)
python字符串过滤性能比较5种方法
2021-09-26 18:35:25
python cookie反爬处理的实现
2021-10-16 23:33:17