随机梯度下降法

代码如下:

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

e0 = 1
e1 = 1
e2 = 1

alpha = 0.01
theta0 = np.random.uniform(0, 1)
theta1 = np.random.uniform(0, 1)
theta2 = np.random.uniform(0, 1)
theta = np.array([theta1, theta2])

eps = 1e-4

x = np.array([[2, 3], [4, 6], [7, 8], [12, 15]])
t = np.array([5, 8, 14, 17])

cnt = 0  # 迭代次数
while (abs(e0) >= eps or abs(e1) >= eps or abs(e2) >= eps):  # 这里用绝对值小于eps进行判定 如果不用绝对值的话如果有数据小于0的话会直接跳出循环
    cnt += 1
    i = np.random.randint(0,4) #随机一个整数作为本次的训练数据
    e0 = (np.sum((x[i] * theta)) + theta0 * 1 - t[i])
    e1 = (np.sum((x[i] * theta)) + theta0 * 1 - t[i]) * x[i][0]
    e2 = (np.sum((x[i] * theta)) + theta0 * 1 - t[i]) * x[i][1]
    i = np.random.randint(0,4)
    e0 += (np.sum((x[i] * theta)) + theta0 * 1 - t[i])
    e1 += (np.sum((x[i] * theta)) &
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值