Skip to content

Instantly share code, notes, and snippets.

@SkalskiP

SkalskiP/keras_add_callback.py

Created September 13, 2018 22:03
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save SkalskiP/6a99acbba953fcfdeb8e29291ffebba5 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save SkalskiP/6a99acbba953fcfdeb8e29291ffebba5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
keras_add_callback.py
# Adding callback functions that will run on each epoch
testmodelcb = keras.callbacks.LambdaCallback(on_epoch_end=save_model_prediction_graph)
# Compilation of the model
model.compile(loss='binary_crossentropy', optimizer='adamax', metrics=['accuracy'])
# Model training
model.fit(X_train, y_train, epochs=N_EPOCHS, verbose=0, callbacks=[testmodelcb])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment