_pickle.UnpicklingError:NEWOBJ类参数不是类型对象

2024-03-29 09:40:23 发布

您现在位置:Python中文网/ 问答频道 /正文

我在cmd上运行了这段代码,我遇到了这个错误。 _pickle.UnpicklingError: NEWOBJ class argument isn't a type object

我的app.py如下所示:

# -*- coding: utf-8 -*-
"""
Created on Wed Apr 21 19:18:42 2021

@author: Agni Sain
"""
from flask import Flask,request,render_template
import pandas as pd
import numpy as np
import pickle

app=Flask(__name__)
with open('xgb.pkl','rb') as f:
    xgb=pickle.load(f)

@app.route('/')    
def home():
    return render_template("index.html")   

    
@app.route('/predict',methods=["POST"])
def predict_heart_disease():
    
    age=request.form['age']
    sex=request.form['sex']
    cp = request.form['cp']
    trestbps = request.form['trestbps']
    chol = request.form['chol']
    fbs = request.form['fbs']
    restecg = request.form['restecg']
    thalach = request.form['thalach']
    exang = request.form['exang']
    oldpeak = request.form['oldpeak']
    slope = request.form['slope']
    ca = request.form['ca']
    thal = request.form['thal']
    
    pvalues=[[age,sex,cp,trestbps,chol,fbs,restecg,thalach,exang,oldpeak,slope,ca,thal]]
    pvalues=np.array(pvalues).reshape((1,-1))
    pred=xgb.predict(pvalues)
    predf=float(pred)
    return render_template('result.html', data=predf)

@app.route('/predict_file',methods=["POST"])    
def predict_heart_disease_file():
    df_test=pd.read_csv(request.files.get("file"))
    
    prediction=xgb.predict(df_test)
    return str((list(prediction)))
    
if (__name__=='__main__'):
    app.run()

我确实在这里读到了这个错误,但是没有从中找到任何这样的帮助。有人能简单地说一下为什么会出现这个错误以及如何解决它吗


Tags: importformappreturnrequestdefas错误
2条回答

您需要添加if语句

df_test=pd.read_csv(request.files.get("file"))
if not df_test.empty:
   prediction=xgb.predict(df_test)
   return str((list(prediction)))
else:
   return None

正如本thread中所讨论的,原因是XGBClassifier直接与scikit-learnAPI交互,因此也需要在远程服务器上安装该API。如果您使用xgboost.booster类(本机xgboost模型,而不是scikit学习类型模型),它就可以正常工作
此外,按照model I/O page进行保存和加载应该可以正常工作。
另一个“答案”不正确,错误并非源于数据帧为空/无

相关问题 更多 >