使用python编写函数梯度下降算法
时间: 2023-05-21 07:04:38 浏览: 131
基于Python实现的梯度下降算法.zip
可以,以下是一个简单的Python函数梯度下降算法的示例:
```
def gradient_descent(cost_func, gradient_func, initial_guess, alpha, max_iter):
guess = initial_guess
for i in range(max_iter):
gradient = gradient_func(guess)
guess = guess - alpha * gradient
cost = cost_func(guess)
if i % 100 == 0:
print("Iteration {}: Cost = {}".format(i, cost))
return guess
```
其中,`cost_func`是代价函数,`gradient_func`是梯度函数,`initial_guess`是初始猜测,`alpha`是学习率,`max_iter`是最大迭代次数。这个函数利用梯度下降算法寻找代价函数的最小值。
阅读全文