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.