boax.core.optimizers.initializers.q_batch

boax.core.optimizers.initializers.q_batch(samples, num_restarts, eta=1.0)

Q batch initializer.

Example

>>> initializer = q_batch(fun, samples, num_restarts)
>>> candidates = initializer(key)

Parameters:

• fun – The scoring function.

• samples (Union[Array, ndarray, bool, number]) – The candidate samples.

• num_restarts (int) – The number of restarts.

• eta (Union[Array, ndarray, bool, number, float, int]) – The temperature parameter.

Return type:

Initializer

Returns:

The q-batch Initializer.