首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于布谷鸟搜索的XGBoost算法优化及应用研究
引用本文:李欣玲,李莉莉,周楷贺.基于布谷鸟搜索的XGBoost算法优化及应用研究[J].青岛大学学报(自然科学版),2023(4):35-40.
作者姓名:李欣玲  李莉莉  周楷贺
作者单位:青岛大学经济学院
基金项目:国家社科基金(批准号:2019BTJ028)资助;
摘    要:为提高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算法得到显著提升。

关 键 词:XGBoost  布谷鸟搜索  分类预测  回归预测
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号