mm1.py

...
    i, j = 0, batch_size
    for _ in range(nbatches):
        sub = df.iloc[i:j]
        X2 = bc[i:j]
        ...
        # Calculate X and Y appropriately
        ...
        yield [X, X2], Y
        i = j
        j += batch_size
        ...