ml.rst

Machine Learning utilities

.. currentmodule:: pyspark.pandas.mlflow

MLflow

Arbitrary MLflow models can be used with Koalas Dataframes, provided they implement the 'pyfunc' flavor. This is the case for most frameworks supported by MLflow (scikit-learn, pytorch, tensorflow, ...). See comprehensive examples in :func:`load_model` for more information.

Note

The MLflow package must be installed in order to use this module. If MLflow is not installed in your environment already, you can install it with the following command:

pip install koalas[mlflow]

.. autosummary::
   :toctree: api/

   PythonModelWrapper
   load_model