easy minibatch

http://iogi.hatenablog.com/entry/split-list-into-sublist
余りがあるときに捨ててくれるのはある意味ありがたい。。。

zip(*[iter(range(15))]*3)
[(0, 1, 2), (3, 4, 5), (6, 7, 8), (9, 10, 11), (12, 13, 14)]

この場合range(15)が大元のデータ(文や単語など)で、3がミニバッチサイズということになる(はず。)

なおpython3ではizipはitertoolsに含まれているので
from itertools import zip_longest
のようにする。

タグ:

python
最終更新:2016年10月19日 09:34