随机梯度下降

随机梯度下降,第1张

随机梯度下降

import numpy as np
import random
h0 = np.random.random()
h1 = np.random.random()
h2 = np.random.random()
h = np.array([h1,h2])
x=np.array([[2014,3],[1600,3],[2400,2],[1416,2],[3000,4]])
t=np.array([400,330,369,232,540])
a = 0.001
eps = 1e-4
e0 = 10
e1 = 10
e2 = 10
i=random.randint(0,4)
while  e0>=eps or e1>=eps or e2>=eps:
    e0 =(np.sum((x[i,0:2]*h[0]))+h0*1-t[i])
    e1 = e0*x[i,0]
    e2 = e0*x[i,1]
    h0 = h0-a*e0
    h1 = h1-a*e1
    h2 = h2-a*e2
print(h0, h1, h2)

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5480713.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-12
下一篇 2022-12-12

发表评论

登录后才能评论

评论列表(0条)

保存