我在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()
我确实在这里读到了这个错误,但是没有从中找到任何这样的帮助。有人能简单地说一下为什么会出现这个错误以及如何解决它吗
您需要添加if语句
正如本thread中所讨论的,原因是
XGBClassifier
直接与scikit-learn
API交互,因此也需要在远程服务器上安装该API。如果您使用xgboost.booster
类(本机xgboost模型,而不是scikit学习类型模型),它就可以正常工作此外,按照model I/O page进行保存和加载应该可以正常工作。
另一个“答案”不正确,错误并非源于数据帧为空/无
相关问题 更多 >
编程相关推荐