code: def train_on_batch(
      self,
      x,
      y=None,
      sample_weight=None,
      class_weight=None,
      reset_metrics=True,
  ):
      return self.model.train_on_batch(
          x=x,
          y=y,
          sample_weight=sample_weight,
          class_weight=class_weight,
          reset_metrics=reset_metrics,
      )
  
imports: import numpy
  from bigdl.optim.optimizer import MaxEpoch
  from zoo.tfpark.utils import evaluate_string_metrics
  from zoo.common import enable_multi_fs_save, enable_multi_fs_load
  from zoo.common.nncontext import getOrCreateSparkContext
  from zoo.tfpark.tf_dataset import TFNdarrayDataset, TFDataset, _standarize_feature_label_dataset, check_data_compatible
  from zoo.tfpark.tf_optimizer import TFOptimizer
  from zoo.tfpark.tf_predictor import TFPredictor
label_desc: Use image node_id 16 for calling the KerasModel obj's underlying member method code with example usage: obj.train_on_batch(x, y, sample_weight, class_weight, reset_metrics) and returns: self