baba_schedule#

statrl.settings.bandits.stochastic.batch.envs.parametric.baba_schedule(horizon, nbArms)[source]#

Batch sizes of the BABA epoch grid, as a plain list.

Parameters:

  • horizon (int) – Number of rounds the grid is built for.

  • nbArms (int) – Number of arms.

Returns:

One batch size per round. A list rather than a callable, so it stays picklable across the process pool used by multicoreRuns().

Return type:

list of int

See also

statrl.settings.bandits.stochastic.batch.agents.baba_schedule.compute_baba_grid

Returns the phase and epoch labels alongside these sizes.