Machine Learning utilities#
MLflow#
Arbitrary MLflow models can be used with pandas-on-Spark Dataframes,provided they implement the ‘pyfunc’ flavor. This is the casefor most frameworks supported by MLflow (scikit-learn, pytorch,tensorflow, …). See comprehensive examples inload_model() for more information.
Note
The MLflow package must be installed in order to use this module.
| A wrapper around MLflow's Python object model. |
| Loads an MLflow model into a wrapper that can be used both for pandas and pandas-on-Spark DataFrame. |
On this page