tfx.components.schema_gen.executor.Executor

View source on GitHub

Generic TFX schema_gen executor.

Inherits From: BaseExecutor

Child Classes

class Context

Methods

Do

View source

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.
  • Returns
    None