XGBoost.predict() TypeError: predict() got an unexpected keyword argument ‘data‘_typeerror: predict_probability() got an unexpected

@创建于：20211126

1. 问题描述

利用XGBoost算法，训练好模型后，开展预测，代码如下：

pred = xgb_model.predict(data=test_X)
1

报错，信息如下：
TypeError: predict() got an unexpected keyword argument ‘data’

2. 解决办法

# 改写代码
# pred = xgb_model.predict(data=test_X)
# 如下方式
pred = xgb_model.predict(test_X)
1
2
3
4

3. 原因

# XGBoost==1.3.3的关键字是 data
pred = xgb_model.predict(data=test_X)
# XGBoost==1.5.0的关键字是 X
pred = xgb_model.predict(X=test_X)
1
2
3
4

在xgboost的sklearn.py中，不同版本的接口是不一样的。

3.1 XGBoost==1.3.3的predict()接口

3.2 XGBoost==1.5.0的predict()接口