How to register 2 ML flow models(encoder model and xg-boost/or any model) under one model name

Question

How to register 2 ML flow models(encoder model and xg-boost/or any model) under one model name

ShilpaGopal 105

How can I register onehot encoder pickle file and the xg-boost model under a single model name using MLFlow?(or what is the right approach to handle encoding and then model training)

mlflow.register_model(model_uri, args.model_name) - this registers single model under model name

I am able to dump 2 pickle file under one model as below:

User's image

Using MLflow, how can I register 2 models under one model name?

User's image

Ramr-msft 17,826 Reputation points

2023-05-19T11:20:17.9933333+00:00

@ShilpaGopal Thanks for the question. you can register multiple versions of a model under the same model name using MLflow. From the MLflow Runs detail page, select a logged MLflow Model in the Artifacts section. Click the Register Model button. In the Model Name field, if you are adding a new model, specify a unique name to identify the model. If you are registering a new version to an existing model, pick the existing model name from the dropdown.
ShilpaGopal 105 Reputation points

2023-05-19T18:38:56.0166667+00:00

@Ramr-msft thanks for the suggestion. I want to register 2 models under same version . The link says that if a model with provided name already exists then it creates a new version of same model. My requirement is to register 2 modes under the same version.

1 answer

Your answer

Ramr-msft 17,826 Reputation points

2023-05-19T11:20:17.9933333+00:00

@ShilpaGopal Thanks for the question. you can register multiple versions of a model under the same model name using MLflow. From the MLflow Runs detail page, select a logged MLflow Model in the Artifacts section. Click the Register Model button. In the Model Name field, if you are adding a new model, specify a unique name to identify the model. If you are registering a new version to an existing model, pick the existing model name from the dropdown.
ShilpaGopal 105 Reputation points

2023-05-19T18:38:56.0166667+00:00

@Ramr-msft thanks for the suggestion. I want to register 2 models under same version . The link says that if a model with provided name already exists then it creates a new version of same model. My requirement is to register 2 modes under the same version.

Answer 1

Ramr-msft 17,826

@ShilpaGopal Thanks for the details. Currently it's not supported, we suggest you to use the model versioning.

Filiz Camuz 0 Reputation points

2024-02-15T23:57:44.89+00:00

Is there any update for this issue in azureml studio?

Share via

How to register 2 ML flow models(encoder model and xg-boost/or any model) under one model name

1 answer

Your answer