You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: articles/machine-learning/how-to-use-mlflow-azure-databricks.md
+11-11Lines changed: 11 additions & 11 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,7 +1,7 @@
1
1
---
2
-
title: MLflow tracking for Azure Databricks ML experiments
2
+
title: MLflow tracking for Azure Databricks machine learning experiments
3
3
titleSuffix: Azure Machine Learning
4
-
description: Set up MLflow with Azure Machine Learning to log metrics and artifacts from Azure Databricks ML experiments.
4
+
description: Set up MLflow with Azure Machine Learning to log metrics and artifacts from Azure Databricks machine learning experiments.
5
5
services: machine-learning
6
6
author: msakande
7
7
ms.author: mopeakande
@@ -14,7 +14,7 @@ ms.custom: sdkv2
14
14
#customer intent: As a data scientist, I want to integrate Azure Databricks with Azure Machine Learning to connect the products.
15
15
---
16
16
17
-
# Track Azure Databricks ML experiments with MLflow and Azure Machine Learning
17
+
# Track Azure Databricks machine learning experiments with MLflow and Azure Machine Learning
18
18
19
19
[MLflow](https://www.mlflow.org) is an open-source library for managing the life cycle of your machine learning experiments. You can use MLflow to integrate Azure Databricks with Azure Machine Learning to ensure you get the best from both of the products.
20
20
@@ -59,7 +59,7 @@ You can configure Azure Databricks to track experiments using MLflow in two ways
59
59
60
60
By default, when you link your Azure Databricks workspace, dual-tracking is configured for you.
61
61
62
-
### Dual-tracking on Azure Databricks and Azure Machine Learning
62
+
### Dual-track on Azure Databricks and Azure Machine Learning
63
63
64
64
Linking your Azure Databricks workspace to your Azure Machine Learning workspace enables you to track your experiment data in the Azure Machine Learning workspace and Azure Databricks workspace at the same time. This configuration is called *Dual-tracking*.
65
65
@@ -96,7 +96,7 @@ with mlflow.start_run():
96
96
> [!NOTE]
97
97
> As opposed to tracking, model registries don't support registering models at the same time on both Azure Machine Learning and Azure Databricks. For more information, see [Registering models in the registry with MLflow](#registering-models-in-the-registry-with-mlflow).
98
98
99
-
### Tracking exclusively on Azure Machine Learning workspace
99
+
### Track exclusively on Azure Machine Learning workspace
100
100
101
101
If you prefer to manage your tracked experiments in a centralized location, you can set MLflow tracking to *only* track in your Azure Machine Learning workspace. This configuration has the advantage of enabling easier path to deployment using Azure Machine Learning deployment options.
102
102
@@ -233,19 +233,19 @@ After you configure tracking, configure how to authenticate to the associated wo
When you configure MLflow to exclusively track experiments in Azure Machine Learning workspace, the experiment naming convention has to follow the one used by Azure Machine Learning. In Azure Databricks, experiments are named with the path to where the experiment is saved, forinstance `/Users/[email protected]/iris-classifier`. However,in Azure Machine Learning, you provide the experiment name directly. The same experiment would be named `iris-classifier` directly.
After this configuration, you can use MLflow in Azure Databricks in the same way as you're used to. For more information, see [Log & view metrics and log files](how-to-log-view-metrics.md).
247
247
248
-
## Logging models with MLflow
248
+
## Log models with MLflow
249
249
250
250
After your model is trained, you can log it to the tracking server with the `mlflow.<model_flavor>.log_model()` method. `<model_flavor>` refers to the framework associated with the model. [Learn what model flavors are supported](https://mlflow.org/docs/latest/models.html#model-api).
251
251
@@ -262,7 +262,7 @@ Models are logged inside of the run being tracked. That fact means that models a
262
262
> [!IMPORTANT]
263
263
> The parameter `registered_model_name` has not been specified. For more information about this parameter and the registry, see [Registering models in the registry with MLflow](#registering-models-in-the-registry-with-mlflow).
264
264
265
-
## Registering models in the registry with MLflow
265
+
## Register models in the registry with MLflow
266
266
267
267
As opposed to tracking, model registries can't operate at the same timein Azure Databricks and Azure Machine Learning. They have to use either one or the other. By default, model registries use the Azure Databricks workspace. If you choose to [set MLflow tracking to only track in your Azure Machine Learning workspace](#tracking-exclusively-on-azure-machine-learning-workspace), the model registry is the Azure Machine Learning workspace.
- If a registered model with the name doesn’t exist, the method registers a new model, creates version 1, and returns a `ModelVersion` MLflow object.
277
277
- If a registered model with the name already exists, the method creates a new model version and returns the version object.
278
278
279
-
### Using Azure Machine Learning Registry with MLflow
279
+
### Use Azure Machine Learning registry with MLflow
280
280
281
281
If you want to use Azure Machine Learning Model Registry instead of Azure Databricks, we recommend that you [set MLflow tracking to only track in your Azure Machine Learning workspace](#tracking-exclusively-on-azure-machine-learning-workspace). This approach removes the ambiguity of where models are being registered and simplifies the configuration.
For a complete example of this scenario, see [Training models in Azure Databricks and deploying them on Azure Machine Learning](https://github.com/Azure/azureml-examples/blob/main/sdk/python/using-mlflow/deploy/track_with_databricks_deploy_aml.ipynb).
293
293
294
-
## Deploying and consuming models registered in Azure Machine Learning
294
+
## Deploy and consume models registered in Azure Machine Learning
295
295
296
296
Models registered in Azure Machine Learning Service using MLflow can be consumed as:
0 commit comments