cancel
Showing results forย 
Search instead forย 
Did you mean:ย 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results forย 
Search instead forย 
Did you mean:ย 

How to PREVENT mlflow's autologging from logging ALL runs?

naveen_marthala
Contributor

I am logging runs from jupyter notebook. the cells which has `mlflow.sklearn.autlog()` behaves as expected. but, the cells which has .fit() method being called on sklearn's estimators are also being logged as runs without explicitly mentioning `mlflow.sklearn.autlog()` on top. How do I have mlflow log only the ones I call `mlflow.xxxx.autlog()` or by doing `with mlflow.star_run()`?

1 ACCEPTED SOLUTION

Accepted Solutions

Anonymous
Not applicable
3 REPLIES 3

Anonymous
Not applicable

Anonymous
Not applicable

Joe_Breath1
New Contributor III

It looks like MLflow auto-logging is kicking in by default whenever you call .fit(), which is why youโ€™re seeing runs even without explicitly using mlflow.sklearn.autolog(). To fix this, you can disable the global autologging and only trigger it when you explicitly call mlflow.xxx.autolog() or wrap your code with with mlflow.start_run(). For more details, you can also visit website resources on MLflowโ€™s official docs.