教程:Azure Databricks 上的端到端 ML 模型

本教程笔记本演示了在 Azure Databricks 中训练模型的端到端示例,包括加载数据、可视化数据、设置并行超参数优化,以及使用 MLflow 查看结果、注册模型,以及使用 Spark UDF 中的已注册模型对新数据执行推理。

你可以导入此笔记本并自己运行,也可以复制代码片段和想法供自己使用。

笔记本

如果为 Unity Catalog 启用了工作区,请使用此版本的笔记本:

在 Databricks 上将 scikit-learn 与 MLflow 集成配合使用 (Unity Catalog)

获取笔记本

如果未为 Unity Catalog 启用工作区,请使用此版本的笔记本:

在 Databricks 上将 scikit-learn 与 MLflow 集成配合使用

获取笔记本