This commit is contained in:
Jonas Weinz 2018-07-03 15:41:39 +02:00
parent cffa49eade
commit abd61c35d1
2 changed files with 2 additions and 2 deletions

View File

@ -144,7 +144,7 @@
{
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "a4899ee1720f4db4a136a96657f3283a",
"model_id": "d304cda50752491da1637b292a9367e8",
"version_major": 2,
"version_minor": 0
},

View File

@ -641,7 +641,7 @@ class trainer(object):
named_steps[k].fit = lambda X, y: named_steps[k].train_on_batch(to_dense_if_sparse(X), y) # ← why has keras no sparse support on batch progressing!?!?!
if batch_size is None:
self.pm.fit(X = self.sdm.X[:max_size], y = self.sdm.y[:max_size], epochs=n_epochs)
self.pm.fit(X = self.sdm.X[:max_size], y = self.sdm.y[:max_size])
else:
n = len(self.sdm.X) // batch_size
for i in range(n_epochs):