@创建于:20211126
1. 问题描述
利用XGBoost算法,训练好模型后,开展预测,代码如下:
pred = xgb_model.predict(data=test_X)
报错,信息如下:
TypeError: predict() got an unexpected keyword argument ‘data’
2. 解决办法
# 改写代码
# pred = xgb_model.predict(data=test_X)
# 如下方式
pred = xgb_model.predict(test_X)
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)
在xgboost的sklearn.py中,不同版本的接口是不一样的。