cancel
Showing results for 
Search instead for 
Did you mean: 
Machine Learning
cancel
Showing results for 
Search instead for 
Did you mean: 

Can I save MLflow artifacts to locations other than the dbfs?

User16826992666
Valued Contributor

The default location or MLflow artifacts is on dbfs, but I would like to save my models to an alternative location. Is this supported, and if it is how can I accomplish it?

1 ACCEPTED SOLUTION

Accepted Solutions

sajith_appukutt
Honored Contributor II

You could mount an s3 bucket in the workspace and save your model using the mounts DBFS path

For e.g

modelpath = "/dbfs/my-s3-bucket/model-%f-%f" % (alpha, l1_ratio)
mlflow.sklearn.save_model(lr, modelpath)

View solution in original post

1 REPLY 1

sajith_appukutt
Honored Contributor II

You could mount an s3 bucket in the workspace and save your model using the mounts DBFS path

For e.g

modelpath = "/dbfs/my-s3-bucket/model-%f-%f" % (alpha, l1_ratio)
mlflow.sklearn.save_model(lr, modelpath)

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.