code: def test_data_creator(config, batch_size):
      dataset = tfds.load(
          "cats_vs_dogs",
          split=["train[:40%]", "train[40%:50%]", "train[50%:60%]"],
          as_supervised=True,
      )[2]
      dataset = dataset.map(preprocess).batch(batch_size)
      return dataset
  
file_path: BigDL/python/orca/tutorial/tf/transfer_learning.py
imports: import math
  import tensorflow
  import tensorflow_datasets
  from tensorflow import keras
  from tensorflow.keras import layers
  from bigdl.orca import init_orca_context, stop_orca_context
  from bigdl.orca.learn.tf2 import Estimator
label_desc: Use image node_id 5 for calling a global function with example usage: test_data_creator(config, batch_size) and returns: dataset