使用 Databricks 和 MLflow 进行机器学习模型训练和部署的应用实践| 学习笔记(一)

本文涉及的产品
模型训练 PAI-DLC,5000CU*H 3个月
交互式建模 PAI-DSW,5000CU*H 3个月
模型在线服务 PAI-EAS,A10/V100等 500元 1个月
简介: 快速学习使用 Databricks 和 MLflow 进行机器学习模型训练和部署的应用实践

开发者学堂课程【Databricks数据洞察公开课使用 Databricks 和 MLflow 进行机器学习模型训练和部署的应用实践】学习笔记,与课程紧密联系,让用户快速学习知识。

课程地址:https://developer.aliyun.com/learning/course/1058/detail/15565


使用 Databricks 和 MLflow 进行机器学习模型训练和部署的应用实践

 

内容介绍:

一、    学习工作流中的一些痛点问题

二、    Mlflow 的核心功能

三、    Mlflow + DDI 环境介绍

 

这次课程中会介绍如何使用 Databricks Datainsight Mlflow 进行机器学习模型的训练和部署,首先会对机器学习工作中的痛点做一个简单的总结,然后介绍 Mlflow 的主要组件以及这些组件的功能,接着会介绍如何使用 DDIMlflow 来搭建机器学习平台,以管理机器学习的一个生命周期,最后会一个简单的案例来演示 DDI Mlflow ,实现从数据准备模型训练到模型部署和测试的整个的流程。


一、学习工作流中的一些痛点问题


首先,是很难对机器学习的时间进行追踪,本次课程的学习算法中有大量的配置参数,在做机器学习实验的时候,其实很难追踪到哪些参数,哪个版本的代码,以及哪个版本的数据会产生特定的结果,其次是进行机器实验的结果是很难复现的,因为没有标准的方式来打包环境,即使是相同的代码,相同参数以及相同的数据可能也很难复现实验结果。

因为实验结果还取决于采用的代码库,重要的是没有一个标准的方式来管理模型的生命周期,一个算法团队通常会创建大量的模型,这些模型其实是很需要一个中央平台来进行管理的,特别是模型的版本所属阶段以及注释这些元数据信息,和这个版本的模型是由什么代码、数据和参数产生的,以及这个模型的性能指标是怎样的,重要的就是没有一个统一的方式来部署这些模型。


Mlflow 的核心功能


Mlflow 就是为了解决机器学习工作流中的上述痛点问题而生的

Mlflow 可以通过简单的 API 实现实验参数追踪、环境打包、模型管理及模型部署的整个流程。


1.Mlflow Tracking

要介绍的第一个 Mlflow 的核心功能就是 Mlflow Tracking

Mlflow Tracking 去追踪机器学习的实验参数、模型的性能指标以及模型的各种文件,在做机器学习实验的时候,通常需要记录一些参数配置以及模型的性能指标,那么 Mlflow 可以帮助用户免去手动记录的操作,并且 Mlflow 不仅仅支持记录参数还能记录任意的文件,包括了模型、图片、源码等等。

image.png

从上图的代码可以看到,我们使用 Mlflowstart-run 可以开启一次实验,然后可以使用 log-param 去记录模型的参数配置,接着可以使用 log-metrics 记录下模型的性能指标,这里的性能指标指标,不但包括标量的性能指标,而且也可以去记录一些向量的性能指标,最后,可以使用 log-model 记录训练好的模型以及使用 log-artifact 去记录任何想要记录的文件,比如上图记录的就是源码。

 

2.Mlflow Project

Mlflow 会基于代码规约来打包训练代码,并且指定执行环境进行入口,以及参而且这种规范的打包方式,更加方便代码的共享以及平台的迁移。

image.png

比如在上图 Mlflow -training Project 里包含有有两个很重要的文件,一个是conda.yaml 文件,一个是 MLProject文件,在conda.yaml 这个文件中指定了project 的一个运行环境,包括了其所有的依赖的代码库以及这些代码库的一个版本。

 image.png

如上图所示,在 MLproject 里,指定了运行的环境,也就是conda的环境,然后,还指定了运行的入口,就是怎么把这个Project run起来。

mflow run\

-no-conda—experiment-namewine-quality\http://github.com/Mlflow/Mlflow-example.git\-P alpha=0.5

如以上代码所示,而在这个入口信息里,又有相应的一个运行参数,这里包含了 alpha 还有 l1-ratio 这两个参数,除此之外, Mlflow 还提供了命令行工具来方便的运行 Mlflow project

比如已经打包好了一个Project ,并且把这个 Project 已经上传到了 git 的仓库里,上传到了 Mlflow-example.git  这个仓库中,只需要执行 Mlflow run ,就可以去方便的去执行这个 Project - p 是去传入 alpha 的一个参数。


3.  Mlflow Models

它支持以统一的方式去打包,记录和部署多种算法框架的模型。

image.png

在训练完模型后,可以使用去Mlflowlog Models 把这一模型给记录下来,然后 Mlflow 自动的把这个模型给存储起来,可以存储到本地或者是存储到 OSS 上,我们在 Mlflow 的外部 UI 上就可以看到这个模型与代码版本参数和metric 之间的关系,并且还记录了这一模型的一个存储的路径。

如上图,最后,可以使用 Mlflow 提供的 API 去部署这个模型,使用 Mlflow models serve 把这个模型部署起来,之后就可以使用 REST API 去调用这个模型得到一个预测的结果。


4. Model Registry

如下图, Mlflow 不但会存储模型,而且还提供了一个 Web UI 去管理这些模型,在这个 Web UI 的界面上它展示了模型的一个版本、模型所处的阶段。

image.png

在模型的详情页,如上图,它还显示了模型的描述以及模型的标签。

模型的标签可以用来检索和标记一个模型,还有模型的schema ,模型的 schema 可以表示那个模型的一个输入和输出的一个格式,此外 Mlflow 他还建立了模型以及运行环境代码和参数之间的一个关系,也就是模型的血缘。

Mlflow 这四个个核心功能,很好的解决了机器学习工作流中的一些痛点问题。

首先是 Mlflow Tracking 很好地解决了,机器学习实验难以追踪的一个问题;Mlflow Project 解决了机器学习工作流中,没有标准的方式来打包环境,实验结果,难以复现的问题; Mlflow Models Model Registry解决的是没有标准的方式来管理模型的生命周期的问题。

 

三、 Mlflow + DDI环境介绍


使用 Mlflow DDI 搭建机器学习平台与管理机器学习的生命周期,在以下架构图中可以发现主要组件有DDI 集群、OSS ECS

image.png

DDI 集群中,用户会去做机器学习的训练,然后,用户需要启动一台 ECS来搭建 Mlflow Tracking Server 以提供一个 UI 界面,此外,还需要在 ECS 上安装 MySQL 去存储训练参数、性能和标签等元数据。

然后 OSS 是用作存储训练数据以及模型源码等等,部署的要点是首先需要去打通 DDI 集群到 ECS 的一个网络,要保证 DDI 集群和 ECS 之间能相互通信,其次,需要在 ECS 上安装 Mlflow-aliyunstore 的插件,以便于 ECS 将模型存储到 OSS 上,环境所需要的 DDI 集群, OSS Bucket 以及用作 MySQL ECS 环境都已经准备好了。

首先需要去打通 DDI 集群到 ECS 网络,然后在 DDI 的管控台数据源目录下,

选择添加,如下图所示。

image.png

然后选择 ECS 实例,并且填入名称,点击“下一步”即可打通数据源。

image.png

已经打通过了,就不重复去添加。

然后登录到作为 Mlflow Server ECS 上,首先下载 Anaconda

wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh

image.png

下一步,去安装 Anaconda

bash ./Anaconda3-2021.05-Linux-x86_64.sh


相关实践学习
ECS云服务器新手上路
本实验会自动创建一台ECS实例。首先,远程登陆ECS实例,并部署应用。然后,登陆管理控制台,并对这台ECS实例进行管理操作。
7天玩转云服务器
云服务器ECS(Elastic Compute Service)是一种弹性可伸缩的计算服务,可降低 IT 成本,提升运维效率。本课程手把手带你了解ECS、掌握基本操作、动手实操快照管理、镜像管理等。了解产品详情: https://www.aliyun.com/product/ecs
相关文章
|
24天前
|
机器学习/深度学习 算法 开发工具
通义千问2(Qwen2)大语言模型在PAI-QuickStart的微调、评测与部署实践
阿里云的人工智能平台PAI,作为一站式的机器学习和深度学习平台,对Qwen2模型系列提供了全面的技术支持。无论是开发者还是企业客户,都可以通过PAI-QuickStart轻松实现Qwen2系列模型的微调、评测和快速部署。
|
8天前
|
机器学习/深度学习 人工智能 Java
人工智能平台PAI产品使用合集之已经通过自定义镜像部署了一个模型,想要上传并导入其他模型,该如何操作
阿里云人工智能平台PAI是一个功能强大、易于使用的AI开发平台,旨在降低AI开发门槛,加速创新,助力企业和开发者高效构建、部署和管理人工智能应用。其中包含了一系列相互协同的产品与服务,共同构成一个完整的人工智能开发与应用生态系统。以下是对PAI产品使用合集的概述,涵盖数据处理、模型开发、训练加速、模型部署及管理等多个环节。
|
7天前
|
机器学习/深度学习 存储 分布式计算
PAI机器学习平台如何进行分布式训练?
【7月更文挑战第1天】PAI机器学习平台如何进行分布式训练?
19 1
|
16天前
|
机器学习/深度学习 API 索引
机器学习项目实践-基础知识部分
创建Python隔离环境使用`python -m venv`命令,如`python -m venv ml`来创建名为`ml`的虚拟环境。激活环境通过`.\<Scripts>\activate`(Windows)。然后可以使用`pip`安装库,如`numpy`、`pandas`、`matplotlib`和`jupyter notebook`。在虚拟环境中,`numpy`是用于数组计算的库,支持数学操作和绘图。`pip install`命令后面可添加`-i Simple Index`指定索引源。完成安装后,激活环境并启动`jupyter notebook`进行开发。
|
18天前
|
机器学习/深度学习 数据采集 人工智能
探索机器学习:从理论到实践
【6月更文挑战第19天】本文旨在为读者提供一个机器学习的全面视角,从基本概念到实际应用案例,再到面临的挑战和未来趋势。我们将通过一个实际的项目案例来深入探讨如何将机器学习理论应用于解决现实世界问题,并分享一些在项目实施过程中学到的经验和教训。无论你是机器学习领域的新手还是有经验的从业者,这篇文章都将为你提供有价值的见解和启发。
25 4
|
19天前
|
机器学习/深度学习 存储 人工智能
【机器学习】Samba-CoE实现高效推理部署
【机器学习】Samba-CoE实现高效推理部署
53 5
|
19天前
|
机器学习/深度学习 人工智能 算法
【机器学习】Q-Learning算法:在序列决策问题中的实践与探索
【机器学习】Q-Learning算法:在序列决策问题中的实践与探索
26 0
【机器学习】Q-Learning算法:在序列决策问题中的实践与探索
|
26天前
|
机器学习/深度学习 数据采集 关系型数据库
机器学习入门:使用Scikit-learn进行实践
机器学习入门:使用Scikit-learn进行实践
268 1
|
3天前
|
机器学习/深度学习 传感器 自动驾驶
【机器学习】机器学习赋能交通出行:智能化实践与创新应用探索
【机器学习】机器学习赋能交通出行:智能化实践与创新应用探索
15 0
|
4天前
|
机器学习/深度学习 分布式计算 算法
Java中的机器学习模型集成与训练实践
Java中的机器学习模型集成与训练实践

相关产品

  • 人工智能平台 PAI