gen_batches

sklearn.utils.gen_batches(n, batch_size, *, min_batch_size=0)

Генератор для создания срезов, содержащих batch_size элементы от 0 до n.

Последний срез может содержать меньше чем batch_size элементы, когда batch_size не делит n.

Параметры:

nint

Размер последовательности.

batch_sizeint

Количество элементов в каждой партии.

min_batch_sizeint, по умолчанию=0

Минимальное количество элементов в каждой партии.

Возвращает:

срез batch_size элементы

Смотрите также

gen_even_slices

Генератор для создания n_packs срезов до n.

Примеры

>>> from sklearn.utils import gen_batches
>>> list(gen_batches(7, 3))
[slice(0, 3, None), slice(3, 6, None), slice(6, 7, None)]
>>> list(gen_batches(6, 3))
[slice(0, 3, None), slice(3, 6, None)]
>>> list(gen_batches(2, 3))
[slice(0, 2, None)]
>>> list(gen_batches(7, 3, min_batch_size=0))
[slice(0, 3, None), slice(3, 6, None), slice(6, 7, None)]
>>> list(gen_batches(7, 3, min_batch_size=2))
[slice(0, 3, None), slice(3, 7, None)]