损失函数y=(x-5)**2
def gran_1(x):
# y=(x-5)**2
return 2*(x-5)
def gran_d(current, learn_rate=0.1, min=0.0000001, max_i=100):
for i in range(max_i):
gran=gran_1(current)
if abs(gran) <= min:
return current
current -= learn_rate*gran
print(gran_d(1))