azureml-mlflow 包

mlflow

包含将 Azure 机器学习与 MLFlow 集成的功能。

MLflow (https://mlflow.org/) 是用于跟踪机器学习试验和管理模型的开源平台。 可将 MLflow 日志记录 API 与 Azure 机器学习结合使用,以便将指标和项目记录到 Azure 机器学习工作区。

在 Azure 机器学习 workspace 中,添加以下代码来使用 MLflow。 get_mlflow_tracking_uri 方法将 MLflow 跟踪 URI 设置为指向工作区。

import mlflow
from azureml.core import Workspace
workspace = Workspace.from_config()
mlflow.set_tracking_uri(workspace.get_mlflow_tracking_uri())

可以在 https://aka.ms/azureml-mlflow-examples 找到更多示例。