hotfix
This commit is contained in:
parent
cffa49eade
commit
abd61c35d1
@ -144,7 +144,7 @@
|
||||
{
|
||||
"data": {
|
||||
"application/vnd.jupyter.widget-view+json": {
|
||||
"model_id": "a4899ee1720f4db4a136a96657f3283a",
|
||||
"model_id": "d304cda50752491da1637b292a9367e8",
|
||||
"version_major": 2,
|
||||
"version_minor": 0
|
||||
},
|
||||
|
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user