tfx.components.example_gen.utils.generate_output_split_names

View source on GitHub

Return output split name based on input and output config.

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.