基于消息隊列的LightGBM超參數優化
摘要：為了提高LightGBM超參數優化效率,同時得到全局最優模型,提出了以消息隊列方式并行優化LightGBM超參數方法。根據超參數的預選范圍,將每一組超參數發送到隊列中,各節點從隊列獲取到消息后以并行方式進行模型訓練并驗證準確率,最后選出準確率最高的模型計算待預測的數據集。實驗結果表明,與傳統的網格搜索、貝葉斯法、隨機搜索方法以及消息隊列串行優化相比,消息隊列并行優化超參數方法時間最短,AUC值最大。
注: 保護知識產權,如需閱讀全文請聯系計算機工程與科學雜志社
相關推薦
更多
