摘要:为了提高LightGBM超参数优化效率,同时得到全局最优模型,提出了以消息队列方式并行优化LightGBM超参数方法。根据超参数的预选范围,将每一组超参数发送到队列中,各节点从队列获取到消息后以并行方式进行模型训练并验证准确率,最后选出准确率最高的模型计算待预测的数据集。实验结果表明,与传统的网格搜索、贝叶斯法、随机搜索方法以及消息队列串行优化相比,消息队列并行优化超参数方法时间最短,AUC值最大。
关键词:消息队列 lightgbm 超参数
单位:新疆大学电气工程学院; 新疆乌鲁木齐830047; 国网新疆电力有限公司电力科学研究院; 新疆乌鲁木齐; 830011
注:因版权方要求,不能公开全文,如需全文,请咨询杂志社