Function batch_iterator

mla/utils/main.py:10–25

Splits X into equal sized chunks.

(X, batch_size=64)

def batch_iterator(X, batch_size=64):
    """Splits X into equal sized chunks."""
    n_samples = X.shape[0]
    n_batches = n_samples // batch_size
    batch_end = 0

    for b in range(n_batches):
        batch_begin = b * batch_size
        batch_end = batch_begin + batch_size

        X_batch = X[batch_begin:batch_end]

        yield X_batch

    if n_batches * batch_size < n_samples:
        yield X[batch_end:]