机器学习--Flask 发布一个预测模型

通过 Flask 制作一个简单的微服务预测接口,实现模型的预测和浏览器交互功能:

import joblib
from flask import Flask, request
import numpy as np

'''
# 不使用浏览器,常规的控制台交互
model_path = ".//model.pkl"
model = joblib.load(model_path)
sen = [[['坚决', 'a', 'ad', '1_v'],
        ['惩治', 'v', 'v', '0_Root'],
        ['贪污', 'v', 'v', '1_v'],
        ['贿赂', 'n', 'n', '-1_v'],
        ['等', 'u', 'udeng', '-1_v'],
        ['经济', 'n', 'n', '1_v'],
        ['犯罪', 'v', 'vn', '-2_v']]]
print(model.predict(sen))
'''

# Flask 实现模型和浏览器进行交互
app = Flask(__name__)

@app.route('/')
def hello_world():
    return '你好'

@app.route('/predict/<sen>')
def predict(sen):
    print("参数:", type(sen), sen)  # 得到的为字符串而不是数组,故结果不对,解决办法??? 没想到。。
    result = model.predict(sen)
    return str(result)

if __name__ == '__main__':
    app.config['JSON_AS_ASCII'] = False  # 中文乱码
    model_path = ".//model.pkl"
    model = joblib.load(model_path)
    app.run()

原文:
https://soyoger.blog.csdn.net/article/details/108729394

好的,以下是一个简单的使用 Flask 实现机器学习模型注册网页的示例代码: ```python from flask import Flask, render_template, request import joblib app = Flask(__name__) # 加载训练好的机器学习模型 model = joblib.load('model.pkl') # 定义首页路由 @app.route('/') def index(): return render_template('index.html') # 定义提交表单的路由 @app.route('/register', methods=['POST']) def register(): # 从表单中获取特征值 feature1 = float(request.form['feature1']) feature2 = float(request.form['feature2']) feature3 = float(request.form['feature3']) feature4 = float(request.form['feature4']) # 预处理特征值 features = [[feature1, feature2, feature3, feature4]] # 使用模型预测结果 prediction = model.predict(features) # 渲染结果 HTML 模板 return render_template('result.html', prediction=prediction) if __name__ == '__main__': app.run(debug=True) ``` 以上代码中,我们首先加载了训练好的机器学习模型(使用 joblib 库),然后定义了两个路由:`/` 是首页路由,返回一个 HTML 表单;`/register` 是提交表单的路由,从表单中获取特征值,预处理特征值,使用模型预测结果,最后渲染结果 HTML 模板。 我们还需要创建两个 HTML 模板文件:`index.html` 和 `result.html`,分别用于显示表单和预测结果。这里只给出 `index.html` 的示例代码: ```html <!DOCTYPE html> <html> <head> <meta charset="utf-8"> <title>机器学习模型注册</title> </head> <body> <h1>机器学习模型注册</h1> <form method="post" action="{{ url_for('register') }}"> <p>特征 1:<input type="text" name="feature1"></p> <p>特征 2:<input type="text" name="feature2"></p> <p>特征 3:<input type="text" name="feature3"></p> <p>特征 4:<input type="text" name="feature4"></p> <p><input type="submit" value="提交"></p> </form> </body> </html> ``` 这个模板文件包含一个表单,用于输入特征值。我们可以根据实际情况修改模板文件,添加样式和其他元素。 最后,我们还需要在命令行中运行应用程序: ``` $ export FLASK_APP=app.py $ flask run ``` 运行成功后,可以在浏览器中访问 http://localhost:5000 查看应用程序。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值