带参数的最小化函数

2024-09-28 21:18:49 发布

您现在位置:Python中文网/ 问答频道 /正文

目前我有以下代码定义函数f

a = #something
b = #something
c = #something
def f(x):
    """Evaluates some function that depends on parameters a, b, and c"""
    someNumber = #some calculation
    return someNumber

理想情况下,我会做def f(x, a, b, c),但我正在最小化f相对于x,而且SciPy的优化工具箱不允许使用参数最小化函数。也就是说,我想对多个值a, bc运行我的最小化代码。有办法吗?


Tags: and函数代码that定义ondeffunction
2条回答

这是一个关于使用minimize的简单问答。如果其他用户需要更具体的东西,这里有一个简单的例子。

广义二次方程:

In [282]: def fun(x, a,b,c):
     ...:     return a*x**2 + b*x + c

In [283]: optimize.minimize(fun, 10, args=(1,0,0))
Out[283]: 
      fun: 1.7161984122524196e-15
 hess_inv: array([[ 0.50000001]])
      jac: array([ -6.79528891e-08])
  message: 'Optimization terminated successfully.'
     nfev: 15
      nit: 4
     njev: 5
   status: 0
  success: True
        x: array([ -4.14270251e-08])

In [284]: optimize.minimize(fun, 10, args=(1,1,1))
Out[284]: 
      fun: 0.7500000000000221
 hess_inv: array([[ 0.49999999]])
      jac: array([  3.12924385e-07])
  message: 'Optimization terminated successfully.'
     nfev: 12
      nit: 2
     njev: 4
   status: 0
  success: True
        x: array([-0.49999985])

函数也可以将数组作为输入,但仍需要返回单个(标量)值:

In [289]: optimize.minimize(fun, [10,10,10], args=(np.array([1,2,3]), 1, 1))
Out[289]: 
      fun: 2.541666666667115
 hess_inv: array([[ 0.50021475, -0.00126004,  0.00061239],
       [-0.00126004,  0.25822101, -0.00259327],
       [ 0.00061239, -0.00259327,  0.16946887]])
      jac: array([ -8.94069672e-08,   4.47034836e-07,  -2.20537186e-06])
  message: 'Optimization terminated successfully.'
     nfev: 55
      nit: 9
     njev: 11
   status: 0
  success: True
        x: array([-0.50000006, -0.2499999 , -0.16666704])

In [286]: def fun(x, a,b,c):
 ...:     return (a*x**2 + b*x + c).sum()

最好确保函数使用建议的x0和args运行,例如

In [291]: fun(np.array([10,10,10]), np.array([1,2,3]), 1, 1)
Out[291]: 633

如果不能调用目标函数,或者对其参数的工作方式感到困惑,minimize不是一个神奇的项目符号。这种最小化只和你对目标函数的理解一样好。

可以在args中指定其他参数

from scipy.optimize import minimize 
minimize(f, x0, args=(a, b, c))

相关问题 更多 >