pyspark.pandas.mlflow.PythonModelWrapper��

class pyspark.pandas.mlflow.PythonModelWrapper(model_uri: str, return_type_hint: Union[str, type, numpy.dtype, pandas.core.dtypes.base.ExtensionDtype])[source]��

A wrapper around MLflow���s Python object model.

This wrapper acts as a predictor on pandas-on-Spark

Methods

predict(data)

Returns a prediction on the data.
Machine Learning utilities pyspark.pandas.mlflow.load_model