joblib.load 和 pickle.load 错误“没有属性‘XGBoostLabelEncoder’”

Question

我正在尝试使用 joblib 和 pickle 加载 XBGClassifier 模型。

使用 joblib 时loaded_model = joblib.load('trained_model.sav')出现以下错误：AttributeError: module 'xgboost.compat' has no attribute 'XGBoostLabelEncoder'

使用pickle，loaded_model = pickle.load(open('trained_model.sav', 'rb'))我得到AttributeError: Can't get attribute 'XGBoostLabelEncoder' on <module 'xgboost.compat' from 'C:\Users\Usuario\Anaconda3\lib\site-packages\xgboost\compat.py'>

我再次安装了 xgboost，但它不起作用，可能是什么问题？

Answer 1

确保保存和加载环境中的 xgboost 版本相同。您需要在尝试加载模型的环境中更改 xgboost 的版本。