azure_ml_sdk.utils.mlflow_models¶
Classes¶
Class for querying MLflow models. |
Module Contents¶
- class azure_ml_sdk.utils.mlflow_models.MLFlowModel(client: mlflow.tracking.MlflowClient)¶
Class for querying MLflow models.
- client¶
MLflow tracking client.
- Type:
MLflowClient
- client¶
- list_models()¶
List all registered models’ names.
- Returns:
A list of all registered models.
- check_model(model_name)¶
Check if a model exists.
- Parameters:
model_name (str) – The model name.
- Returns:
A boolean object.
- query_model(model_name, tags)¶
Query all registered version of a model.
- Parameters:
model_name (str) – The model name.
tags (list) – The list of tags.
- Returns:
A DataFrame of the registered model versions.
- find_best_model(model_name, tag)¶
Find the best registered model.
- Parameters:
model_name (str) – The model name.
tag – A tag that can have numerical values.
- Returns:
The version of the best model.
- find_best_value(model_name, tag)¶
Find the best value from the registered models.
- Parameters:
model_name (str) – The model name.
tag – A tag that can have numerical values.
- Returns:
The best value of the registered models.