python - 在Python中加载xgboost模型，该模型由R中的 `xgboost::save()`保存

Question

我有一个 xgboost .model 文件，它是在 R 中使用 xgboost::save() 生成的。现在，我想加载该文件并在 python 中使用它。

Answer 1

这似乎是不可能的(编辑:在标准 python xgboost 库中)，因为 python 实现无法从字节串加载模型，这是一个根据 this github thread 的错误

该线程中的注释提供了使用 xgboost.core 库的解决方法函数:

import ctypes
import xgboost
import xgboost.core

def xgb_load_model(buf):
    if isinstance(buf, str):
        buf = buf.encode()
    bst = xgboost.core.Booster()
    n = len(buf)
    length = xgboost.core.c_bst_ulong(n)
    ptr = (ctypes.c_char * n).from_buffer_copy(buf)
    xgboost.core._check_call(
        xgboost.core._LIB.XGBoosterLoadModelFromBuffer(bst.handle, ptr, length)
    )  
    return bst

如果您在二进制文件中读取类似以下内容:

with open('xgb_model.model','rb') as f:
    raw = f.read()

您应该能够通过以下方式从字节串加载:

model = xgb_load_model(raw)