Join the SIG TFX-Addons community and help make TFX even better!

tfx.components.example_gen.utils.generate_output_split_names

Return output split name based on input and output config.

tfx.components.example_gen.utils.generate_output_split_names(
    input_config: Union[example_gen_pb2.Input, Dict[Text, Any]],
    output_config: Union[example_gen_pb2.Output, Dict[Text, Any]]
) -> List[Text]

Return output split name if it's specified and input only contains one split, otherwise output split will be same as input.

input_config example_gen_pb2.Input instance. If any field is provided as a RuntimeParameter, input_config should be constructed as a dict with the same field names as Input proto message.
output_config example_gen_pb2.Output instance. If any field is provided as a RuntimeParameter, output_config should be constructed as a dict with the same field names as Output proto message.

List of split names.

RuntimeError if configs are not valid, including:

  • Missing field.
  • Duplicated split.
  • Output split is specified while input has more than one split.
  • Missing train and eval split.