Export and import models

To save models, use the MLflow functions log_model and save_model. You can also save models using their native APIs onto Databricks File System (DBFS). For MLlib models, use ML Pipelines.

To export models for serving individual predictions, you can use MLeap, a common serialization format and execution engine for machine learning pipelines. MLeap supports serializing Apache Spark, scikit-learn, and TensorFlow pipelines into a bundle, so you can load and deploy trained models to make predictions with new data. You can import the exported models into both Spark and other platforms for scoring and predictions.

On Databricks Runtime 5.5 LTS you can use Databricks ML Model Export.