作者:骑着母猪去相亲 | 来源:互联网 | 2023-05-19 06:30
我是python和机器学习的新手.我有一个线性回归模型,它能够根据我已经转储用于Web服务的输入来预测输出.请参阅以下代码:X_train,X_test,y_train,y_tes
我是python和机器学习的新手.我有一个线性回归模型,它能够根据我已经转储用于Web服务的输入来预测输出.请参阅以下代码:
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25)
regression_model = LinearRegression()
regression_model.fit(X_train, y_train)
print(regression_model.predict(np.array([[21, 0, 0, 0, 1, 0, 0, 1, 1, 1]]))) # this is returning my expected output
joblib.dump(regression_model, '../trainedModels/MyTrainedModel.pkl')
使用flask我试图将其作为Web服务公开,如下所示:
@app.route('/predict', methods=['POST'])
def predict():
X = [[21, 0, 0, 0, 1, 0, 0, 1, 1, 1]]
model = joblib.load('../trainedModels/MyTrainedModel.pkl')
prediction = model.predict(np.array(X).tolist())
return jsonify({'prediction': list(prediction)})
但它抛出了以下异常
Object of type 'ndarray' is not JSON serializable
我试过NumPy array is not JSON serializable
但仍然是同样的错误.我该如何解决这个问题
解决方法:
尝试使用tolist()方法转换您的ndarray:
prediction = model.predict(np.array(X).tolist()).tolist()
return jsonify({'prediction': prediction})
json包示例:
a = np.array([1,2,3,4,5]).tolist()
json.dumps({"prediction": a})
那应该输出:
'{"prediction": [1, 2, 3, 4, 5]}'