make_batch
make_batch(data: Union[Iterable, Generator], batch_size: int) -> Generator[List, None, None]
-
Description: This function is used to convert data into batches.
-
Parameters
- data (
Union[Iterable, Generator]
): The data generator. - batch_size (
int
): The size of each batch.
- data (
-
Returns
- Generator[List, None, None]: A generator that yields batched data.
-
Example
import capybara as cb
data = range(10)
batch_size = 3
batched_data = cb.make_batch(data, batch_size)
for batch in batched_data:
print(batch)
# >>> [0, 1, 2]
# >>> [3, 4, 5]
# >>> [6, 7, 8]
# >>> [9]