使用梯度下降算法时的初始值误差



问题:初始值为10000,解收敛到10000,而不是实际解1。

import numpy.linalg as nl
x_ini=10000
def obj(x):
f = x**2 - 2*x + 3
return f
def grad(x):
df = 2*x - 2
return df
n_iter=1000
lr=0.001
x_old = x_ini

for _ in range(True):

x_new = x_old - lr*(grad(x_old))
x_old = x_new

if(nl.norm(grad(x_old))<=1e-03):
break
x_new = x_old

print(x_new)
while True:

x_new = x_old - lr*(grad(x_old))
x_old = x_new
y = nl.norm(grad(x_old))
if(y<=1e-03):
break
x_new = x_old
print(x_new)

您可以将for更改为while

最新更新