gen_batches

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

Generator zum Erstellen von Slices, die batch_size Elemente von 0 bis n enthalten.

Der letzte Slice kann weniger als batch_size Elemente enthalten, wenn batch_size n nicht teilt.

Parameter:

nint

Größe der Sequenz.

batch_sizeint

Anzahl der Elemente in jedem Batch.

min_batch_sizeint, Standard=0

Minimale Anzahl von Elementen in jedem Batch.

Gibt:

Slice von batch_size Elementen

Siehe auch

gen_even_slices

Generator zum Erstellen von n_packs Slices bis n.

Beispiele

>>> 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)]