code: def _get_prediction_data(self):
      assert not self.drop_remainder, (
          "sanity check: drop_remainder should be false in this case,"
          " otherwise please report a bug"
      )
      jvalue = callZooFunc(
          "float",
          "createMiniBatchRDDFromTFDataset",
          self.rdd.map(lambda x: x[0]),
          self.init_op_name,
          self.table_init_op,
          self.output_names,
          self.output_types,
          self.shard_index_op_name,
      )
      rdd = jvalue.value().toJavaRDD()
      return rdd
  
imports: import tensorflow
  from zoo.tfpark.tf_dataset import TensorMeta
  from zoo.util import nest
  from zoo import getOrCreateSparkContext, get_node_and_core_number
  from zoo.common import callZooFunc
  from zoo.feature.common import FeatureSet
  from zoo.orca.data import SparkXShards
  from zoo.tfpark import TFDataset
label_desc: Use image node_id 2 for calling the TFDataDataset2 obj's underlying member method code with example usage: obj._get_prediction_data() and returns: rdd