pyspark.pandas.mlflow.PythonModelWrapper#

class pyspark.pandas.mlflow.PythonModelWrapper(model_uri, return_type_hint)[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.