This commit is contained in:
Jonas Weinz 2018-07-03 15:40:04 +02:00
parent 8cbd7cbe90
commit cffa49eade

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], validation_split=0.1, epochs=n_epochs)
self.pm.fit(X = self.sdm.X[:max_size], y = self.sdm.y[:max_size], epochs=n_epochs)
else:
n = len(self.sdm.X) // batch_size
for i in range(n_epochs):