作者： 俊欣
来源：关于数据分析与可视化
对于机器学习爱好者而言，很多时候我们需要将建好的模型部署在线上，实现前后端的交互，今天小编就通过Flask以及Streamlit这两个框架实现机器学习模型的前后端交互。
首先是模型的建立，小编这回为了省事儿建立一个非常简单的二分法模型，所引用的数据集如下所示
import pandas as pd
df = pd.read_csv("data.csv")
df.head()
output
Height Weight Species 0 88.9 48.3 Dog 1 90.2 47.4 Dog 2 82.7 44.8 Dog 3 81.4 48.2 Dog 4 83.5 39.9 Dog
所涉及到的特征也就两列分别是“Hight”以及“Weight”也就是身高和体重，我们需要通过这两个特征来预测它到底是“猫”还是“狗”，代码如下
X = df[["Height", "Weight"]] y = df["Species"]
clf = GaussianNB()
clf.fit(X, y)
当然小编为了省事儿这里并没有进行训练集和测试集的区分，也没有进行任何的调参以及模型的优化，只是简单的建立了一个朴素贝叶斯的二分类模型。接下来我们将建立好的模型保存下来
import joblib
joblib.dump(clf, "clf.pkl")
前端页面主要是由一系列的HTML代码写成的，代码如下
<!DOCTYPE html> <html> <head> <title>Your Machine Learning App</title> </head> <body> <form name="form", method="POST", style="text-align: center;"> <br> Height: <input type="number" name="height", placeholder="Enter height in cm" required/> <br><br> Weight: <input type="number" name="weight", placeholder="Enter weight in kg" required/> <br><br> <button value="Submit">Run</button> </form> <p style="text-align: center;">{{ output }}</p> </body> </html>
输出结果如下：
我们可以看到有两个输入框分别代表的是身高与体重，以及运行的按钮键。接下来我们来写后端的逻辑代码，当前端传过来数据的时候，也就是身高与体重的数据的时候，后端的代码来调用已经训练好的模型并且做出预测，然后显示在前端的页面上。在Flask框架中后端的业务代码大致如下
from flask import Flask, request, render_template import pandas as pd import joblib # 声明是一个Flask应用 app = Flask(__name__) # 主要业务逻辑 # ------------------ # 运行整体的应用 if __name__ == '__main__':
app.run(debug = True)
那么在本篇文章的项目背景下，代码如下
@app.route('/', methods=['GET', 'POST']) def main(): # 表单数据提交，POST请求 if request.method == "POST": # 调用已经训练好的模型 clf = joblib.load("clf.pkl") # 从输入框中获取身高与体重数据 height = request.form.get("height")
weight = request.form.get("weight") # 转变成DataFrame格式 X = pd.DataFrame([[height, weight]], columns = ["Height", "Weight"]) # 获取预测值 prediction = clf.predict(X)[0] else: prediction = "" return render_template("website.html", output = prediction)
然后我们运行整个脚本，效果如下
我们试着输入一些身高与体重的值，看一下返回的结果，效果如下
下面我们来看一下将模型部署在Streamlit框架下该如何来操作。在Streamlit框架中没有特别明显的前后端代码的分离，代码如下
import streamlit as st import pandas as pd import joblib # 标题 st.header("Streamlit Machine Learning App") # 输入框 height = st.number_input("Enter Height")
weight = st.number_input("Enter Weight") # 点击提交按钮 if st.button("Submit"): # 引入训练好的模型 clf = joblib.load("clf.pkl") # 转换成DataFrame格式的数据 X = pd.DataFrame([[height, weight]],
columns=["Height", "Weight"]) # 获取预测出来的值 prediction = clf.predict(X)[0] # 返回预测的值 st.text(f"This instance is a {prediction}")
最后生成的页面如下
我们在终端中运行以下命令
streamlit run streamlit_model.py
最后我尝试在输入框中填入一些虚构的数字，看一下出来的结果是什么样的，如下
数据分析咨询请扫描二维码
若不方便扫码，搜微信号：CDAshujufenxi
在当今这个数据驱动的时代，几乎每一个业务决策都离不开对数据的深入分析。而其中，指标波动归因分析更是至关重要的一环。无论是 ...
2025-02-18当数据开始说谎：那些年我们交过的学费 你有没有经历过这样的场景？熬了三个通宵做的数据分析报告，在会议上被老板一句"这数据靠 ...
2025-02-17数据分析作为一门跨学科领域，融合了统计学、编程、业务理解和可视化技术。无论是初学者还是有一定经验的从业者，系统化的学习路 ...
2025-02-17挖掘用户价值本质是让企业从‘赚今天的钱’升级为‘赚未来的钱’，同时让用户从‘被推销’变为‘被满足’。询问deepseek关于挖 ...
2025-02-17近来deepseek爆火，看看deepseek能否帮我们快速实现数据看板实时更新。 可以看出这对不知道怎么动手的小白来说是相当友好的， ...
2025-02-14一秒精通 Deepseek，不用找教程，不用买资料，更不用报一堆垃圾课程，所有这么去做的，都是舍近求远，因为你忽略了 deepseek 的 ...
2025-02-12自学 Python 的关键在于高效规划 + 实践驱动。以下是一份适合零基础快速入门的自学路径，结合资源推荐和实用技巧： 一、快速入 ...
2025-02-12“我们的利润率上升了，但销售额却没变，这是为什么？” “某个业务的市场份额在下滑，到底是什么原因？” “公司整体业绩 ...
2025-02-08活动介绍 为了助力大家在数据分析领域不断精进技能，我们特别举办本期打卡活动。在这里，你可以充分利用碎片化时间在线学习，让 ...
2025-02-071、闺女，醒醒，媒人把相亲的带来了。 我。。。。。。。 2、前年春节相亲相了40个， 去年春节相亲50个， 祖宗，今年你想相多少个 ...
2025-02-06在数据科学的广阔领域中，统计分析与数据挖掘占据了重要位置。尽管它们常常被视为有关联的领域，但两者在理论基础、目标、方法及 ...
2025-02-05在数据分析的世界里，“对比”是一种简单且有效的方法。这就像两个女孩子穿同一款式的衣服，效果不一样。 很多人都听过“货比三 ...
2025-02-05当我们只有非常少量的已标记数据，同时有大量未标记数据点时，可以使用半监督学习算法来处理。在sklearn中，基于图算法的半监督 ...
2025-02-05考虑一种棘手的情况：训练数据中大部分样本没有标签。此时，我们可以考虑使用半监督学习方法来处理。半监督学习能够利用这些额 ...
2025-02-04一、数学函数 1、取整 =INT(数字) 2、求余数 =MOD(除数，被除数) 3、四舍五入 =ROUND（数字，保留小数位数） 4、取绝对值 =AB ...
2025-02-03作者：CDA持证人 余治国 一般各平台出薪资报告，都会哀嚎遍野。举个例子，去年某招聘平台发布《中国女性职场现状调查报告》， ...
2025-02-02真正的数据分析大神是什么样的呢？有人认为他们能轻松驾驭各种分析工具，能够从海量数据中找到潜在关联，或者一眼识别报告中的数 ...
2025-02-01现今社会，“转行”似乎成无数职场人无法回避的话题。但行业就像座围城：外行人看光鲜，内行人看心酸。数据分析这个行业，近几年 ...
2025-01-31本人基本情况： 学校及专业：厦门大学经济学院应用统计 实习经历：快手数据分析、字节数据分析、百度数据分析 Offer情况：北京 ...
2025-01-3001专家简介 徐杨老师，CDA数据科学研究院教研副总监，主要负责CDA认证项目以及机器学习/人工智能类课程的研发与授课，负责过中 ...
2025-01-29
