使用 python 将参数传递给 Flask 中的模型

Question

我有一个经过训练的模型，我已将其导出为 pickle 文件。我正在尝试在我的 python 文件中使用 pickle 文件，该文件位于烧瓶上运行。但是，由于文件出错，我无法传递参数。当我在我的 jupyter notebook 中使用相同的代码时，会传递参数并给出预测。但是，当涉及到运行时，它会出现以下错误：

Traceback (most recent call last):


File "C:\model1\venv\lib\site-packages\flask\app.py", line 2292, in wsgi_app
    response = self.full_dispatch_request()
  File "C:\model1\venv\lib\site-packages\flask\app.py", line 1815, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "C:\model1\venv\lib\site-packages\flask\app.py", line 1718, in handle_user_exception
    reraise(exc_type, exc_value, tb)
  File "C:\model1\venv\lib\site-packages\flask\_compat.py", line 35, in reraise
    raise value
  File "C:\model1\venv\lib\site-packages\flask\app.py", line 1813, in full_dispatch_request
    rv = self.dispatch_request()
  File "C:\model1\venv\lib\site-packages\flask\app.py", line 1799, in dispatch_request
    return self.view_functions[rule.endpoint](**req.view_args)
  File "C:\model1\hello.py", line 26, in predict
    model = pickle.load(open("prediction.pkl","rb"))
  File "sklearn\neighbors\binary_tree.pxi", line 1152, in sklearn.neighbors.kd_tree.BinaryTree.__setstate__
  File "sklearn\neighbors\binary_tree.pxi", line 235, in sklearn.neighbors.kd_tree.get_memview_ITYPE_1D
ValueError: Buffer dtype mismatch, expected 'ITYPE_t' but got 'long long'

我的hello.py文件如下：

import pickle
import numpy as numpy
from decimal import Decimal
from flask import Flask, request, json

from sklearn.externals import joblib

app = Flask(__name__)

@app.route('/predict', methods=['POST'])
def predict():
    print(request.form)
    features = request.form["features"]

    features = json.loads(features)

    features = numpy.array(features)

    features = features.reshape(1, -1)
    model = pickle.load(open("prediction.pkl","rb"))

    prediction = model.predict(features).tolist()
    print(features)
    print(prediction)
    return json.dumps({"Prediction":prediction})

if __name__ == '__main__':
    app.run(host='127.0.0.1')

任何人都可以建议如何解决错误吗？

Answer 1

您好，我认为您的模型已经在不同的 sklearn 库版本上进行了训练，并且加载模型 sklearn 库的版本不同。或 python 版本问题。

使用 python 将参数传递给 Flask 中的模型

Passing parameters to model in Flask using python

python

flask

sklearn-pandas