class_docstr: Custom callbacks for `predict`.
class_docstr_tok: ["Custom","callbacks","for","`","predict","`","."]
code: def on_predict_batch_end(self, batch, logs=None):
      logging.log_every_n(
          logging.INFO,
          "End `predict` on batch %d.",
          _LOG_EVERY_N_BATCHES,
          batch,
      )
  
imports: import multiprocessing
  import os
  import time
  from typing import Any
  from absl import flags
  from absl import logging
  import numpy
  import tensorflow
  from deepvariant import dv_utils
  from deepvariant import keras_modeling
  from deepvariant import logging_level
  from deepvariant.protos import deepvariant_pb2
  from absl import app
  import multiprocessing
  from third_party.nucleus.io import sharded_file_utils
  from third_party.nucleus.io import tfrecord
  from third_party.nucleus.protos import variants_pb2
  from third_party.nucleus.util import errors
  from third_party.nucleus.util import proto_utils
  from third_party.nucleus.util import variant_utils
label_desc: Use image node_id 2 for calling the CustomCallback obj's underlying member method code with example usage: obj.on_predict_batch_end(batch, logs) without return types