apache_beam.ml.inference.sklearn_inference module¶

class apache_beam.ml.inference.sklearn_inference.ModelFileType[source]¶

Bases: enum.Enum

An enumeration.

PICKLE = 1¶

JOBLIB = 2¶

class apache_beam.ml.inference.sklearn_inference.SklearnInferenceRunner[source]¶

Bases: apache_beam.ml.inference.base.InferenceRunner

run_inference(batch: List[numpy.ndarray], model: Any) → Iterable[apache_beam.ml.inference.api.PredictionResult][source]¶

get_num_bytes(batch: List[numpy.ndarray]) → int[source]¶: Returns the number of bytes of data for a batch.

class apache_beam.ml.inference.sklearn_inference.SklearnModelLoader(model_file_type: apache_beam.ml.inference.sklearn_inference.ModelFileType = <ModelFileType.PICKLE: 1>, model_uri: str = '')[source]¶

Bases: apache_beam.ml.inference.base.ModelLoader

load_model()[source]¶: Loads and initializes a model for processing.

get_inference_runner() → apache_beam.ml.inference.sklearn_inference.SklearnInferenceRunner[source]¶