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])¶ 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.