cancel
Showing results forย 
Search instead forย 
Did you mean:ย 
Machine Learning
Dive into the world of machine learning on the Databricks platform. Explore discussions on algorithms, model training, deployment, and more. Connect with ML enthusiasts and experts.
cancel
Showing results forย 
Search instead forย 
Did you mean:ย 

How can I view the storage space taken by a registered model using MLFlow?

fa
New Contributor III

The information viewed about the registered models on the Models tab is very minimal. Just showing the tags we pass in and version information. How can I get more details about the model such as the size on disk?

2 REPLIES 2

Wunlung
Contributor

The registered model actually links with a source run in the mlflow experiment and get those information from that experiment.

A. Using the UI.

  1. Select the model under Registered Models
  2. Select the version which you want to get more detail
  3. Click the link Near the Source Run and redirect you to the experiment page
  4. You will get all the artifacts including model object(size, path, ...), input_example.json, yaml, requirement.txt, ...

B. Use the MLFlowClient

from pprint import pprint
 
client = MlflowClient()
for rm in client.search_registered_models():
    pprint(dict(rm), indent=4)

You will get the source path and can list those objects.

Octavian1
Contributor

Hi,

I have used the MLFlow client, but I am not sure where to find the size of the model image.
The response to 

client.search_registered_models()

 I am getting is the following:


<RegisteredModel: aliases={}, creation_timestamp=17061..., description='', last_updated_timestamp=17061..., latest_versions=None, name='<catalog>.<db>.<model>', tags={}>
{ 'aliases': {},
'creation_timestamp': 17078...,
'description': '',
'last_updated_timestamp': 17078...,
'latest_versions': None,
'name': '<catalog>.<db>.<model>',
'tags': {}}

If going to the catalog UI view, where the model and its versions details are given, there is no such information.

Join 100K+ Data Experts: Register Now & Grow with Us!

Excited to expand your horizons with us? Click here to Register and begin your journey to success!

Already a member? Login and join your local regional user group! If there isn’t one near you, fill out this form and we’ll create one for you to join!