https://community.databricks.com/t5/machine-learning/what-can-i-do-to-reduce-the-number-of-mlflow-api-calls-i-make/m-p/32041#M1687 <P>I'm fitting multiple models in parallel. For each one, I'm logging lots of params and metrics to MLflow. I'm hitting rate limits, causing problems in my jobs.</P> Sun, 09 Jan 2022 01:10:38 GMT Joseph_B 2022-01-09T01:10:38Z https://community.databricks.com/t5/machine-learning/what-can-i-do-to-reduce-the-number-of-mlflow-api-calls-i-make/m-p/32041#M1687 <P>I'm fitting multiple models in parallel. For each one, I'm logging lots of params and metrics to MLflow. I'm hitting rate limits, causing problems in my jobs.</P> Sun, 09 Jan 2022 01:10:38 GMT https://community.databricks.com/t5/machine-learning/what-can-i-do-to-reduce-the-number-of-mlflow-api-calls-i-make/m-p/32041#M1687 Joseph_B 2022-01-09T01:10:38Z https://community.databricks.com/t5/machine-learning/what-can-i-do-to-reduce-the-number-of-mlflow-api-calls-i-make/m-p/32042#M1688 <P>The first thing to try is to log in batches. If you are logging each param and metric separately, you're making 1 API call per param and 1 per metric. Instead, you should use the batch logging APIs; e.g. use "log_params" instead of "log_param" <A href="https://mlflow.org/docs/latest/python_api/mlflow.html#mlflow.log_params" target="test_blank">https://mlflow.org/docs/latest/python_api/mlflow.html#mlflow.log_params</A></P><P>If you're logging 10 params and metrics per model, this will cut the number of API calls you're making by a factor of 10.</P><P></P><P>If this optimization is still insufficient for you, then I'd recommend doing 2 things:</P><P>1) Short-term workaround: You can save data you wish to log to a table and log it in a follow-up process later.</P><P>2) Medium/long-term: I'd recommend working with your Databricks account team to come up with a solution to match your needs. In general, the best option is to reorganize how models are being fit or logged to do more efficient batching in logging. But different cases may have different best solutions, so working with your account team may be the best option.</P> Sun, 09 Jan 2022 01:15:50 GMT https://community.databricks.com/t5/machine-learning/what-can-i-do-to-reduce-the-number-of-mlflow-api-calls-i-make/m-p/32042#M1688 Joseph_B 2022-01-09T01:15:50Z