tfx_bsl.public.beam.RunInference

Run inference with a model.

There are two types of inference you can perform using this PTransform:

  1. In-process inference from a SavedModel instance. Used when saved_model_spec field is set in inference_spec_type.
  2. Remote inference by using a service endpoint. Used when ai_platform_prediction_model_spec field is set in inference_spec_type.

  3. Bytes as Input.

  4. PTable Input.

  5. Models as SideInput.

examples A PCollection containing examples.
inference_spec_type Model inference endpoint.

A PCollection containing prediction logs.