medkit.core.utils#

Functions:

batch_iter(iter, batch_size)

Group values yielded by an iterator into batches.

batch_list(list, batch_size)

Split list into smaller batches.

batch_iter(iter, batch_size)[source]#

Group values yielded by an iterator into batches.

Parameters
  • iter (Iterator[Any]) – The iterator yielding values to batch.

  • batch_size (int) – Length of batches (the last batch may be smaller).

Return type

Iterator[List[Any]]

Returns

Iterator[List[Any]] – Iterator yielding lists of batch_size items (the last list yielded may be smaller).

batch_list(list, batch_size)[source]#

Split list into smaller batches.

Parameters
  • list (List[Any]) – The list containing values to batch.

  • batch_size (int) – Length of batches (the last batch may be smaller).

Return type

Iterator[List[Any]]

Returns

Iterator[List[Any]] – Iterator yielding lists of batch_size items (the last list yielded may be smaller).