python – ‘ndarray’类型的对象不是JSON可序列化的

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – ‘ndarray’类型的对象不是JSON可序列化的，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1151字，纯文字阅读大概需要2分钟。

内容图文

我是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]}'

内容总结

以上是互联网集市为您收集整理的python – ‘ndarray’类型的对象不是JSON可序列化的全部内容，希望文章能够帮你解决python – ‘ndarray’类型的对象不是JSON可序列化的所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---