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: 'statistics': A list of 'ExampleStatistics' type which must contain split 'train'.
output_dict	Output dict from key to a list of artifacts, including: schema: 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