seldonian.parse_tree.zhat_funcs.batcher

batcher(func, N, batch_size, num_batches)

Calls func num_batches times, batching up the inputs.

Parameters:
  • func – The function you want to call

  • N (int) – The total number of datapoints

  • batch_size (int) – The size of each batch

  • num_batches (int) – The number of batches

Returns:

A wrapper function that does the actual function calls