【问题标题】:How to find out the version of the xgboost library on which the model was built如何找出构建模型的 xgboost 库的版本
【发布时间】:2022-12-23 01:48:45
【问题描述】:

我有一个问题:假设我有一个 xgboost 模型,有人训练它并将其放入 .pickle 格式。我要使用这个模型,我需要找出这个模型建立在哪个版本的 xgboost 上。我怎样才能做到这一点?

【问题讨论】:

    标签: python xgboost


    【解决方案1】:

    pickle 文件包含原生 XGBoost 数据格式(可以是二进制或 JSON)的 XGBoost 模型对象。例如,xgboost.sklearn.XGBClassifier 类在其 _Booster 属性中包含 XGBoost 对象。

    XGBoost 对象在其标头区域包含 major_versionminor_version 整数字段。这应该是您想要的信息。

    TLDR:考虑使用以下方法:

    1. 从 pickle 文件中提取 XGBoost 模型对象。
    2. XGBoost 模型对象是以二进制还是 JSON 数据格式编码的?如果是 JSON,那么您正在处理 XGBoost >= 1.3 版本。
    3. 解析 XGBoost 模型对象,并从其标头中提取 major_versionminor_version 字段。

    【讨论】:

      【解决方案2】:

      xgb_model.save_config() 将内部参数配置输出为 JSON。

      您需要找到“版本”属性。

      【讨论】:

        【解决方案3】:

        将 xgboost 导入为 xgb

        xgb。版本

        【讨论】:

          猜你喜欢
          • 1970-01-01
          • 2010-11-23
          • 2022-01-11
          • 2019-04-01
          • 1970-01-01
          • 1970-01-01
          • 2021-11-16
          • 2010-09-21
          • 1970-01-01
          相关资源
          最近更新 更多