南东亮1,2,王维庆1,王海云1
NAN Dong-liang1,2,WANG Wei-qing1,WANG Hai-yun1
摘要:
为了提高LightGBM超参数优化效率,同时得到全局最优模型,提出了以消息队列方式并行优化LightGBM超参数方法。根据超参数的预选范围,将每一组超参数发送到队列中,各节点从队列获取到消息后以并行方式进行模型训练并验证准确率,最后选出准确率最高的模型计算待预测的数据集。实验结果表明,与传统的网格搜索、贝叶斯法、随机搜索方法以及消息队列串行优化相比,消息队列并行优化超参数方法时间最短,AUC值最大。