apache_beam.ml.inference package

A package with various modules for running inferences and predictions on models. This package contains support for popular frameworks as well as an interface for adding unsupported frameworks.

Submodules

• apache_beam.ml.inference.base module
• apache_beam.ml.inference.pytorch_inference module
• apache_beam.ml.inference.sklearn_inference module