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

How to find best model using python in mlflow

User16826994223
Honored Contributor III

I have a use case in mlflow with python code to find a model version that has the best metric (for instance, โ€œaccuracyโ€) among so many versions , I don't want to use web ui but to use python code to achieve this. Any Idea?

1 ACCEPTED SOLUTION

Accepted Solutions

User16826994223
Honored Contributor III
import mlflow 
client = mlflow.tracking.MlflowClient()
runs = client.search_runs("my_experiment_id", "", order_by=["metrics.rmse DESC"], max_results=1)
best_run = runs[0]

https://mlflow.org/docs/latest/python_api/mlflow.tracking.html#mlflow.tracking.MlflowClient.search_r...

View solution in original post

1 REPLY 1

User16826994223
Honored Contributor III
import mlflow 
client = mlflow.tracking.MlflowClient()
runs = client.search_runs("my_experiment_id", "", order_by=["metrics.rmse DESC"], max_results=1)
best_run = runs[0]

https://mlflow.org/docs/latest/python_api/mlflow.tracking.html#mlflow.tracking.MlflowClient.search_r...

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.