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]