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.
Args |
|---|
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.
|
Returns |
|---|
|
List of split names.
|
Raises |
|---|
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.
|
View source on GitHub