示例示例Python求函数极值的方法有很多种,以梯度下降法为例,给出一个代码示例:上述代码中,我们定义了一个函数f(x)=sin(x),然后定义了一个梯度下降求极值的函数grad_descent,设定了学习率lr和步数step_num,最后调用梯度下降求极值函数,传入初始值init_x,即可求出函数f(x)的极值。
Python求函数极值的方法有很多种,以梯度下降法为例,给出一个代码示例:
python
import numpy as np
# 定义函数
def f(x):
return np.sin(x)
# 定义梯度下降求极值函数
def grad_descent(f, init_x, lr=0.01, step_num=100):
x = init_x
for i in range(step_num):
grad = np.gradient(f(x))
x -= lr * grad
return x
# 调用梯度下降求极值函数
init_x = 3 # 设定初始值
grad_descent(f, init_x)
上述代码中,我们定义了一个函数f(x)=sin(x),然后定义了一个梯度下降求极值的函数grad_descent,设定了学习率lr和步数step_num,最后调用梯度下降求极值函数,传入初始值init_x,即可求出函数f(x)的极值。
本站系公益性非盈利分享网址,本文来自用户投稿,不代表边看边学立场,如若转载,请注明出处
评论列表(36条)