Skip to content

Latest commit

 

History

History
42 lines (32 loc) · 1.37 KB

ml.rst

File metadata and controls

42 lines (32 loc) · 1.37 KB
Raw

Machine Learning utilities

.. currentmodule:: pyspark.pandas.mlflow

MLflow

Arbitrary MLflow models can be used with pandas-on-Spark 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.

.. autosummary::
   :toctree: api/

   PythonModelWrapper
   load_model