基于布谷鸟搜索的XGBoost算法优化及应用研究

作者：李欣玲; 李莉莉^*; 周楷贺

来源：青岛大学学报(自然科学版), 2023, 36(04): 35-40.

• XGBoost
• 布谷鸟搜索
• 分类预测
• 回归预测

为提高XGBoost算法预测精度，采用布谷鸟搜索算法全局优化XGBoost的超参数包括学习率、输出结点分裂的最小损失、树模型的最大深度和弱学习器的数量，构建CS-XGBoost模型训练数据集。实验结果表明，基于CS-XGBoost的收入分类模型的准确率、精确率、F1分数和AUC等指标分别为95.67%、97.17%、95.56%和97.96%,均优于Logistic回归、支持向量机、随机森林、XGBoost算法和基于网格搜索的XGBoost算法；基于CS-XGBoost的房价预测模型的决定系数、均方根误差及平均绝对误差分别为0.905 5、2.943 5及2.165 4,预测精度较XGBoost算法得到显著提升。

• 单位
  青岛大学