guides/writing_your_own_callbacks.py/CustomCallback/on_predict_batch_end
class CustomCallback:    def on_predict_batch_end(self, batch, logs=None):
        keys = list(logs.keys())
        print(
            "...Predicting: end of batch {}; got log keys: {}".format(
                batch, keys
            )
        )