Registration is open for TensorFlow Dev Summit 2020 Learn more

tfx.components.bulk_inferrer.executor.Executor

View source on GitHub

Class Executor

TFX bulk inferer executor.

Inherits From: BaseExecutor

__init__

View source

__init__(context=None)

Constructs a beam based executor.

Child Classes

class Context

Methods

Do

View source

Do(
    input_dict,
    output_dict,
    exec_properties
)

Runs batch inference on a given model with given input examples.

Args:

  • input_dict: Input dict from input key to a list of Artifacts.
    • examples: examples for inference.
    • model: exported model.
    • model_blessing: model blessing result
  • output_dict: Output dict from output key to a list of Artifacts.
    • output: bulk inference results.
  • exec_properties: A dict of execution properties.
    • model_spec: JSON string of bulk_inferrer_pb2.ModelSpec instance.
    • data_spec: JSON string of bulk_inferrer_pb2.DataSpec instance.

Returns:

None