python拟合曲线（小批量随机梯度下降）

深度学习书籍参考：https://tangshusen.me/Dive-into-DL-PyTorch/#/read_guide

import numpy as np
from matplotlib import pyplot as plt


plt.ion()#开启交互
X=np.linspace(-1,2,1000)
Y=X**3-X**2+X+0.5+np.random.rand(1000)*0.001 #生成样本+噪声

w=np.zeros(4)#需要拟合的函数系数

lr=0.01
for i in range(5000):
    index=np.random.choice(1000,8)
    x=X[index]
    y=Y[index]

    y_pred=w[0]*x**3+w[1]*x**2+w[2]*x+w[3]
    dy_pred=y_pred-y
    
    w[0]=w[0]-lr*np.dot(x**3,dy_pred)/8
    w[1]=w[1]-lr*np.dot(x**2,dy_pred)/8
    w[2]=w[2]-np.dot(x, dy_pred)/8
    w[3]=w[3]-lr*dy_pred.sum()/8
    if i%100==0:
        plt.clf()
        Y_pred=w[0]*X**3+w[1]*X**2+w[2]*X+w[3]
        plt.scatter(X,Y)
        plt.plot(X,Y_pred, c='r')
        plt.pause(0.001)
print(w)#打印拟合出的函数的系数

代码效果：