функция обратного вызова выдает ошибку в классе scipy.optimize.minimize

Question

Я использую scipy.minimize для задачи оптимизации.

Это мой код

import numpy as np
from scipy.optimize import minimize
from scipy.optimize import Bounds

#bounds = Bounds([25, 36], [26, 38],[10,27],[6,28],[0,1800],[0,800],[0,100],[25,60],[2,7])

bounds = Bounds([20,6,20,23],[35,9,50,26])

energy_history = []
x_values = []

def objective(x):
    return (-0.20859863*x[0:1] -1.5088649*x[1:2] +0.10707853*x[2:3] +1.6829923*x[3:4] -0.008870916*x[0:1]*x[1:2] + 0.0007393111*x[0:1]*x[2:3] +0.010610705*x[0:1]*x[3:4] + 0.005123541*x[1:2]*x[2:3] +  0.086458616*x[1:2]*x[3:4] -0.007695199*x[2:3]*x[3:4] + 0.00016993227*x[0:1]*x[0:1] -0.026582083*x[1:2]*x[1:2]  + 0.00014467833*x[2:3]*x[2:3] -0.051599417*x[3:4]*x[3:4] - 9.540932)

def callback(x):
    fobj = objective(x)
    x_values.append(x)
    energy_history.append(fobj)


x0 = np.array([34,8,49,25])
res = minimize(objective, x0, method='trust-constr',
               options={'verbose': 1}, bounds=bounds,callback=callback)



optimal_values= res.x

print('optimal values found: ' + str(res.x))
print('energy consumed: ' + str(res.fun))

Я получаю сообщение об ошибке при запуске этого.

Ошибка с функцией обратного вызова:

TypeError: callback() takes 1 positional argument but 2 were given

Куда я иду не так?

Answer 1

Согласно документам сигнатура функции обратного вызова зависит от выбранного решателя.(Что не очень хорошо.)

Хотя это callback(x) для всех остальных, в вашем случае это callback(x, status), так как вы используете method='trust-constr'.

Просто добавьте этот дополнительный параметр вваш (и игнорируйте его, если вы не используете информацию о статусе).