Batches of Examples — medium Python for AI problem | Incognition

Batches of Examples

10 pts · 10 coins

Python for AIMedium

Batches of Examples

Models are usually fed data a batch at a time rather than all at once.

Task: write a generator function batches(items, size) that hands out lists of size items, in order.

The last batch may be shorter if the items don't divide evenly.
Never hand out an empty batch: an empty items produces nothing at all.
size is at least 1.

Examples

list(batches([1, 2, 3, 4, 5], 2))

returns [[1, 2], [3, 4], [5]]

list(batches([], 3))

returns []

Hints

solution.py

1def batches(items, size):
2    # your code here
3    pass
4