calc_batches_better.py

def calc_batches(train_min: int, train_max: int, train_every: int, n: int) -> dict:
    batches = dict()
    batch = 0
    for row in range(1,n+1):
        if row < train_min:
            pass
        elif row == train_min:
            batches[batch] = dict(start=0, end=row)
        elif row % train_every == 0:
            batch += 1
            batches[batch] = dict(start=max(0,row-train_max), end=row)
    return batches

calc_batches(train_min=1000, train_max=5000, train_every=500, n=10000)

"""
{0: {'start': 0, 'end': 1000},
 1: {'start': 0, 'end': 1500},
 2: {'start': 0, 'end': 2000},
 3: {'start': 0, 'end': 2500},
 4: {'start': 0, 'end': 3000},
 5: {'start': 0, 'end': 3500},
 6: {'start': 0, 'end': 4000},
 7: {'start': 0, 'end': 4500},
 8: {'start': 0, 'end': 5000},
 9: {'start': 500, 'end': 5500},
 10: {'start': 1000, 'end': 6000},
 11: {'start': 1500, 'end': 6500},
 12: {'start': 2000, 'end': 7000},
 13: {'start': 2500, 'end': 7500},
 14: {'start': 3000, 'end': 8000},
 15: {'start': 3500, 'end': 8500},
 16: {'start': 4000, 'end': 9000},
 17: {'start': 4500, 'end': 9500},
 18: {'start': 5000, 'end': 10000}}
"""