OiO.lk Community platform!

Oio.lk is an excellent forum for developers, providing a wide range of resources, discussions, and support for those in the developer community. Join oio.lk today to connect with like-minded professionals, share insights, and stay updated on the latest trends and technologies in the development field.
  You need to log in or register to access the solved answers to this problem.
  • You have reached the maximum number of guest views allowed
  • Please register below to remove this limitation

Return the list of all registered models in MLflow

  • Thread starter Thread starter Carlo
  • Start date Start date
C

Carlo

Guest
I'm using mlflow on databricks and I've trained some models, that I can see in the "registered models" page.

Is there a way to extract the list of these models in code?

Something like

Code:
import mlflow
mlflow.set_tracking_uri("databricks")
model_infos = mlflow.tracking.MlflowClient().some_method_to_list_registered_models() 

for model_info in model_infos:
    print(f"Name: {model_info.name}, Version: {model_info.latest_versions[0].version}")
<p>I'm using mlflow on databricks and I've trained some models, that I can see in the "registered models" page.</p>
<p>Is there a way to extract the list of these models in code?</p>
<p>Something like</p>
<pre><code>import mlflow
mlflow.set_tracking_uri("databricks")
model_infos = mlflow.tracking.MlflowClient().some_method_to_list_registered_models()

for model_info in model_infos:
print(f"Name: {model_info.name}, Version: {model_info.latest_versions[0].version}")
</code></pre>
 
Top