code: def input_fn(
      tf_transform_output, transformed_examples_pattern, batch_size
  ):
      """An input function reading from transformed data, converting to model input.
  
      Args:
        tf_transform_output: Wrapper around output of tf.Transform.
        transformed_examples_pattern: Base filename of examples.
        batch_size: Batch size.
  
      Returns:
        The input data for training or eval, in the form of k.
      """
      return tf.data.experimental.make_batched_features_dataset(
          file_pattern=transformed_examples_pattern,
          batch_size=batch_size,
          features=tf_transform_output.transformed_feature_spec(),
          reader=tf.data.TFRecordDataset,
          label_key=common.LABEL_KEY,
          shuffle=True,
      ).prefetch(tf.data.experimental.AUTOTUNE)
  
file_path: transform/examples/census_example_v2.py
imports: import math
  import os
  import pprint
  import tempfile
  from absl import logging
  import tensorflow
  import tensorflow_transform
  import census_example_common
  from tensorflow_transform.keras_lib import tf_keras
label_desc: Use image node_id 1 for calling a global function with example usage: input_fn(tf_transform_output, transformed_examples_pattern, batch_size) and returns: tf