def METHOD_NAME(include_label=True):
  """Input schema definition.
  Args:
    include_label: Indicates whether the label feature should be included.
  Returns:
    A `Schema` object.
  """
  result = {}
  if include_label:
    result['clicked'] = tf.io.FixedLenFeature(shape=[], dtype=tf.int64)
  for name in _INTEGER_COLUMN_NAMES:
    result[name] = tf.io.VarLenFeature(dtype=tf.int64)
  for name in _CATEGORICAL_COLUMN_NAMES:
    result[name] = tf.io.VarLenFeature(dtype=tf.string)
  return result