lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix shuffle=False in .fit()

Browse Source
This commit is contained in:
fchollet 2015-05-14 11:06:56 -07:00
parent f596d7a57f
commit c8a104b8fa
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
keras/models.py
View File

@ -182,10 +182,7 @@ class Sequential(object):
batches = make_batches(len(X), batch_size)
for batch_index, (batch_start, batch_end) in enumerate(batches):
if shuffle:
batch_ids = index_array[batch_start:batch_end]
else:
batch_ids = slice(batch_start, batch_end)
batch_ids = index_array[batch_start:batch_end]
seen += len(batch_ids)
X_batch = X[batch_ids]
y_batch = y[batch_ids]
@ -329,3 +326,5 @@ class Sequential(object):
weights = [g['param_{}'.format(p)] for p in range(g.attrs['nb_params'])]
self.layers[k].set_weights(weights)
f.close()