databricks.koalas.mlflow.PythonModelWrapper¶
-
class
databricks.koalas.mlflow.PythonModelWrapper(model_uri, return_type_hint)[source]¶ A wrapper around MLflow’s Python object model.
This wrapper acts as a predictor on koalas
-
__init__(model_uri, return_type_hint)[source]¶ Initialize self. See help(type(self)) for accurate signature.
Methods
__init__(model_uri, return_type_hint)Initialize self.
predict(data)Returns a prediction on the data.
-