Train a PySpark model and save in MLeap format


The managed MLflow integration with Databricks on Google Cloud requires Databricks Runtime for Machine Learning 8.1 or above.

The following notebook trains a PySpark model and saves it in MLeap format.

MLflow PySpark model training notebook

Open notebook in new tab