MLflow:使用 python 代码查找具有最佳指标的模型版本
MLflow: find model version with best metric using python code
我正在尝试使用 API 工作流程(python 代码)在多个模型版本中找到具有最佳指标(例如,“准确性”)的模型版本。我知道我们可以使用 web UI 来做到这一点,但我很乐意编写 python 代码来实现这一点。有人可以帮助我吗?
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]
我正在尝试使用 API 工作流程(python 代码)在多个模型版本中找到具有最佳指标(例如,“准确性”)的模型版本。我知道我们可以使用 web UI 来做到这一点,但我很乐意编写 python 代码来实现这一点。有人可以帮助我吗?
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]