Join the SIG TFX-Addons community and help make TFX even better! Join SIG TFX-Addons


A TFX component to do batch inference on a model with unlabelled examples.

BulkInferrer consumes examples data and a model, and produces the inference results to an external location as PredictionLog proto.

BulkInferrer will infer on validated model.


  # Uses BulkInferrer to inference on examples.
  bulk_inferrer = BulkInferrer(

Component outputs contains:

See the BulkInferrer guide for more details.

examples A Channel of type standard_artifacts.Examples, usually produced by an ExampleGen component. required
model A Channel of type standard_artifacts.Model, usually produced by a Trainer component.
model_blessing A Channel of type standard_artifacts.ModelBlessing, usually produced by a ModelValidator component.
data_spec bulk_inferrer_pb2.DataSpec instance that describes data selection.
model_spec bulk_inferrer_pb2.ModelSpec instance that describes model specification.
output_example_spec bulk_inferrer_pb2.OutputExampleSpec instance, specify if you want BulkInferrer to output examples instead of inference result.

outputs Component's output channel dict.



Add per component Beam pipeline args.

beam_pipeline_args List of Beam pipeline args to be added to the Beam executor spec.

the same component itself.