gen_batches

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

產生器，用於建立包含從 0 到 n 的 batch_size 個元素的切片。

當 batch_size 無法整除 n 時，最後一個切片可能包含少於 batch_size 個元素。

參數:

nint

序列的大小。

batch_sizeint

每個批次中的元素數量。

min_batch_sizeint，預設值=0

每個批次中的最小元素數量。

產生：

大小為 batch_size 的元素切片

另請參閱

產生均勻切片

產生器，用於建立最多到 n 的 n_packs 個切片。

範例

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