【发布时间】:2022-12-23 01:48:45
【问题描述】:
我有一个问题:假设我有一个 xgboost 模型,有人训练它并将其放入 .pickle 格式。我要使用这个模型,我需要找出这个模型建立在哪个版本的 xgboost 上。我怎样才能做到这一点?
【问题讨论】:
我有一个问题:假设我有一个 xgboost 模型,有人训练它并将其放入 .pickle 格式。我要使用这个模型,我需要找出这个模型建立在哪个版本的 xgboost 上。我怎样才能做到这一点?
【问题讨论】:
pickle 文件包含原生 XGBoost 数据格式(可以是二进制或 JSON)的 XGBoost 模型对象。例如,xgboost.sklearn.XGBClassifier 类在其 _Booster 属性中包含 XGBoost 对象。
XGBoost 对象在其标头区域包含 major_version 和 minor_version 整数字段。这应该是您想要的信息。
TLDR:考虑使用以下方法:
major_version 和 minor_version 字段。
【讨论】:
xgb_model.save_config() 将内部参数配置输出为 JSON。
您需要找到“版本”属性。
【讨论】:
将 xgboost 导入为 xgb
xgb。版本
【讨论】: