epochs = 10 history = model.fit(train_generator, steps_per_epoch=len(train_generator), epochs=epochs,