cancel
Showing results for 
Search instead for 
Did you mean: 
Data Engineering
cancel
Showing results for 
Search instead for 
Did you mean: 

When using MLflow should I use log_model or save_model?

User16826992666
Valued Contributor

They seem to have similar functions. What is the recommended pattern here?

1 ACCEPTED SOLUTION

Accepted Solutions

sajith_appukutt
Honored Contributor II

mlflow.<model-type>.log_model(model, ...) saves the model to the MLflow tracking server.

mlflow.<model-type>.save_model(model, modelpath) saved the model locally to  a DBFS path.

More details at https://docs.databricks.com/applications/mlflow/models.html

View solution in original post

1 REPLY 1

sajith_appukutt
Honored Contributor II

mlflow.<model-type>.log_model(model, ...) saves the model to the MLflow tracking server.

mlflow.<model-type>.save_model(model, modelpath) saved the model locally to  a DBFS path.

More details at https://docs.databricks.com/applications/mlflow/models.html

Welcome to Databricks Community: Lets learn, network and celebrate together

Join our fast-growing data practitioner and expert community of 80K+ members, ready to discover, help and collaborate together while making meaningful connections. 

Click here to register and join today! 

Engage in exciting technical discussions, join a group with your peers and meet our Featured Members.