Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfx.components.bulk_inferrer.executor.Executor

View source on GitHub

TFX bulk inferer executor.

Inherits From: BaseExecutor

tfx.components.bulk_inferrer.executor.Executor(
    context=None
)

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