使用 MLflow 跟踪 scikit-learn 模型训练

本笔记本基于 MLflow scikit-learn 糖尿病教程

该笔记本演示如何使用 MLflow 跟踪模型训练过程,包括记录模型 parameters、指标、模型本身和其他项目,例如绘图到 Azure Databricks 托管跟踪服务器。 它还包括有关在 MLflow 跟踪 UI 中查看记录的结果的说明。

若要使用 Mosaic AI 模型服务部署训练的模型,请参阅使用 Mosaic AI 模型服务部署模型

MLflow scikit-learn 模型训练笔记本

获取笔记本