popt.cost_functions.rosenbrock

Rosenbrock objective function.

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