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

How to Log Pickle files as a part of Mlflow experiment run

User16826994223
Honored Contributor III

I want to log certain artifacts as python pickle as part of mlflow experiment

Is there a way to achieve this?

1 REPLY 1

sean_owen
Honored Contributor II
Honored Contributor II

Sure, pickle the object to a local file. Log it to your current run with mlflow.log_artifact. That's it. MLflow lets you log just about anything you want. However if you're experimenting with different variations on a sklearn Pipeline model, you could simply create and try each variation, log each directly with mlflow.sklearn.log_model, into one single MLflow experiment.

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.