Question

A data scientist has logged 200 runs to a single MLflow experiment while tuning a gradient boosting model. Each run logged a metric named 'val_rmse' and a parameter named 'max_depth'. The scientist now wants to programmatically retrieve, into a pandas DataFrame, only the runs where max_depth equals 5, sorted so the lowest val_rmse appears first. Which approach accomplishes this?

Accepted Answer

Use mlflow.search_runs() with filter_string="params.max_depth = '5'" and order_by=["metrics.val_rmse ASC"]

Answer

Use mlflow.list_runs() with filter_string="max_depth == 5" and sort_by="val_rmse"

Answer

Use mlflow.search_runs() with filter_string="metrics.max_depth = 5" and order_by=["params.val_rmse ASC"]

Answer

Use mlflow.get_run() in a loop over all run IDs, then filter and sort the resulting list manually with no filter string