Python sklearn gen_batches用法及代碼示例

本文簡要介紹python語言中 sklearn.utils.gen_batches 的用法。

用法:

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

生成器創建包含 batch_size 元素的切片，從 0 到 n。

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

參數：

n：int

batch_size：int

每批中的元素數。

min_batch_size：整數，默認=0

最小批量生產。

生成(Yield)：

batch_size 元素的切片

例子：

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