Function levy

mbd/blackbox/mbd_opt.py:44–54

(X)

def levy(X):
    X = x_min + (x_max - x_min) * (X + 1.0) / 2.0  # map to [-5, 10]
    w = 1.0 + (X - 1.0) / 4.0
    part1 = jnp.sin(jnp.pi * w[..., 0]) ** 2
    part2 = jnp.sum(
        (w[..., :-1] - 1.0) ** 2
        * (1.0 + 10.0 * jnp.sin(jnp.pi * w[..., :-1] + 1.0) ** 2),
        axis=-1,
    )
    part3 = (w[..., -1] - 1.0) ** 2 * (1.0 + jnp.sin(2.0 * jnp.pi * w[..., -1]) ** 2)
    return part1 + part2 + part3


eval_fn = {