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...
