加急见刊

基于消息队列的LightGBM超参数优化

南东亮; 王维庆; 王海云 新疆大学电气工程学院; 新疆乌鲁木齐830047; 国网新疆电力有限公司电力科学研究院; 新疆乌鲁木齐; 830011

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

注: 保护知识产权,如需阅读全文请联系计算机工程与科学杂志社