popt.cost_functions.rosenbrock
Rosenbrock objective function.
def
rosenbrock(state, *args, **kwargs):
5def rosenbrock(state, *args, **kwargs): 6 """ 7 Rosenbrock: http://en.wikipedia.org/wiki/Rosenbrock_function 8 """ 9 x = state[0]['vector'] 10 x0 = x[:-1] 11 x1 = x[1:] 12 f = sum((1 - x0) ** 2) + 100 * sum((x1 - x0 ** 2) ** 2) 13 return f
Rosenbrock: http://en.wikipedia.org/wiki/Rosenbrock_function