xgboost 中 model.predict 与 model.get_booster()预测结果为什么不一样呢？

最新推荐文章于 2023-04-28 10:57:24 发布

yangqingqing0924

最新推荐文章于 2023-04-28 10:57:24 发布

阅读量4k

点赞数 1

分类专栏：机器学习文章标签：机器学习

本文链接：https://blog.csdn.net/weixin_44494773/article/details/105263004

版权

本文探讨了xgboost中model.predict与model.get_booster().predict预测结果不一致的问题。原因在于model.predict在调用get_booster().predict前会使用DMatrix处理数据，包括设置缺失值，而直接调用get_booster().predict未处理缺失值。通过在DMatrix中设置缺失值后，两个方法的预测结果变得一致。

摘要由CSDN通过智能技术生成

xgb model.predict 与 model.get_booster() 方法预测结果为什么不一样呢？

亲测0.71版本训练出来的模型，用0.90版本的xgb预测，结果不一致
同一版本下model.get_booster().predict与model.predict预测结果为什么不一致呢?

import xgboost as xgb
from xgboost.sklearn import XGBRegressor
import xgboost
import pandas as pd
import numpy as np
from sklearn import datasets
from xgboost import DMatrix
boston_data = datasets