Simple callback in Keras

keras_simple_callback.py

# Sometimes EarlyStopping is too much hassle

class myCallback(tf.keras.callbacks.Callback):
    def on_epoch_end(self, epoch, logs={}):
        if (logs.get('acc') >= 0.998):
            print("\nReached 99.8% accuracy so cancelling training!")
            self.model.stop_training = True
cb = myCallback()