cma_es_fmin_rosenbrock_function_minimization_quickstart.py

1import cma
2
3# The fmin function is the high-level interface for the CMA-ES algorithm
4# It minimizes the objective function (cma.ff.rosen) starting from 
5# an initial solution ([0]*10) with an initial step size (sigma=0.5)
6res = cma.fmin(cma.ff.rosen, 10 * [0], 0.5)
7
8# Display the best solution found and its function value
9print("Best solution:", res[0])
10print("Best function value:", res[1])