https://community.databricks.com/t5/data-engineering/how-to-find-best-model-using-python-in-mlflow/m-p/22835#M15715 <PRE><CODE>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]</CODE></PRE><P></P><P><A href="https://mlflow.org/docs/latest/python_api/mlflow.tracking.html#mlflow.tracking.MlflowClient.search_runs" alt="https://mlflow.org/docs/latest/python_api/mlflow.tracking.html#mlflow.tracking.MlflowClient.search_runs" target="_blank">https://mlflow.org/docs/latest/python_api/mlflow.tracking.html#mlflow.tracking.MlflowClient.search_runs</A></P> Fri, 18 Jun 2021 10:48:01 GMT User16826994223 2021-06-18T10:48:01Z https://community.databricks.com/t5/data-engineering/how-to-find-best-model-using-python-in-mlflow/m-p/22834#M15714 <P>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?</P><P></P> Fri, 18 Jun 2021 10:47:20 GMT https://community.databricks.com/t5/data-engineering/how-to-find-best-model-using-python-in-mlflow/m-p/22834#M15714 User16826994223 2021-06-18T10:47:20Z https://community.databricks.com/t5/data-engineering/how-to-find-best-model-using-python-in-mlflow/m-p/22835#M15715 <PRE><CODE>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]</CODE></PRE><P></P><P><A href="https://mlflow.org/docs/latest/python_api/mlflow.tracking.html#mlflow.tracking.MlflowClient.search_runs" alt="https://mlflow.org/docs/latest/python_api/mlflow.tracking.html#mlflow.tracking.MlflowClient.search_runs" target="_blank">https://mlflow.org/docs/latest/python_api/mlflow.tracking.html#mlflow.tracking.MlflowClient.search_runs</A></P> Fri, 18 Jun 2021 10:48:01 GMT https://community.databricks.com/t5/data-engineering/how-to-find-best-model-using-python-in-mlflow/m-p/22835#M15715 User16826994223 2021-06-18T10:48:01Z