From afd37333ff7c2256e895758e1d50d03d5826246a Mon Sep 17 00:00:00 2001 From: Pedro Pei Date: Fri, 12 Jan 2018 08:50:49 +0800 Subject: [PATCH] Fix bugs in data_utls.batch_gen Previous batch_gen will generate the same batch all the time.Now it can generate batches in order and you will get a new batch when you run batch_gen.__next__() --- data_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data_utils.py b/data_utils.py index 101a343..ec932cb 100644 --- a/data_utils.py +++ b/data_utils.py @@ -28,9 +28,9 @@ def split_dataset(x, y, ratio = [0.7, 0.15, 0.15] ): def batch_gen(x, y, batch_size): # infinite while while True: - for i in range(0, len(x), batch_size): + for i in range(0, len(x)): if (i+1)*batch_size < len(x): - yield x[i : (i+1)*batch_size ].T, y[i : (i+1)*batch_size ].T + yield x[i * batch_size: (i+1)*batch_size ].T, y[i * batch_size: (i+1)*batch_size ].T ''' generate batches, by random sampling a bunch of items