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

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

Answer 1

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