Skip to content

Instantly share code, notes, and snippets.

@aniruddha27

aniruddha27/model_fit_generator.py

Last active August 10, 2020 17:49
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save aniruddha27/12b9994b0dcdddba55b448f7e7b0f1de to your computer and use it in GitHub Desktop.
Save aniruddha27/12b9994b0dcdddba55b448f7e7b0f1de to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_fit_generator.py
# Epochs
epochs = 25
# Batch size
batch_size = 32
history = model.fit(train_datagen.flow(x_train,y_train,
batch_size=batch_size,
seed=27,
shuffle=False),
epochs=epochs,
steps_per_epoch=x_train.shape[0] // batch_size,
validation_data=(x_test,y_test),
verbose=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment