tft.get_transform_input_columns

Return columns that are required inputs of TransformDataset.

tft.get_transform_input_columns(
    preprocessing_fn: Callable[[Mapping[str, common_types.TensorType]], Mapping[str, common_types
        .TensorType]],
    specs: Mapping[str, Union[common_types.FeatureSpecType, tf.TypeSpec]],
    force_tf_compat_v1: bool = False
) -> List[str]

Args

---

preprocessing_fn: A tf.transform preprocessing_fn. specs: A dict of feature name to tf.TypeSpecs. If force_tf_compat_v1 is True, this can also be feature specifications. force_tf_compat_v1: (Optional) If True, use Tensorflow in compat.v1 mode. Defaults to False.

Returns

---

A list of columns that are required inputs of the transform tf.Graph defined by preprocessing_fn.