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}} | |
""" |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment