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.