29 lines
723 B
ReStructuredText
29 lines
723 B
ReStructuredText
|
.. _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
|