tfx.components.schema_gen.executor.Executor

Generic TFX schema_gen executor.

Inherits From: BaseExecutor

tfx.components.schema_gen.executor.Executor(
    context: Optional[tfx.dsl.components.base.base_executor.BaseExecutor.Context] = None
)

Child Classes

class Context

Methods

Do

View source

Do(
    input_dict: Dict[Text, List[types.Artifact]],
    output_dict: Dict[Text, List[types.Artifact]],
    exec_properties: Dict[Text, Any]
) -> None

TensorFlow SchemaGen executor entrypoint.

This infers the schema using tensorflow_data_validation on the precomputed stats of 'train' split.

Args
input_dict Input dict from input key to a list of artifacts, including:

  • 'stats': A list of 'ExampleStatistics' type which must contain split 'train'. Stats on other splits are ignored.
  • 'statistics': Synonym for 'stats'.
output_dict Output dict from key to a list of artifacts, including:
  • output: A list of 'Schema' artifact of size one.
    • exec_properties A dict of execution properties, includes:
  • infer_feature_shape: Whether or not to infer the shape of the feature.
  • exclude_splits: Names of splits that will not be taken into consideration when auto-generating a schema.

    • Returns
    None