Skip to content

Instantly share code, notes, and snippets.

@lucidfrontier45

lucidfrontier45/batch_iter.py

Last active July 21, 2016 02:26
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save lucidfrontier45/18e54438eb36fe8d2501 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save lucidfrontier45/18e54438eb36fe8d2501 to your computer and use it in GitHub Desktop.
Download ZIP
batch iterator
Raw
batch_iter.py
from itertools import islice, chain
def batches(iterable, batchsize=10):
it = iter(iterable)
while True:
batch = islice(it, batchsize)
yield chain([next(batch)], batch)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment