.. _api.ml:

==========================
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