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

基于改进网格搜索算法的随机森林参数优化
引用本文:温博文,董文瀚,解武杰,马 骏. 基于改进网格搜索算法的随机森林参数优化[J]. 计算机工程与应用, 2018, 54(10): 154-157. DOI: 10.3778/j.issn.1002-8331.1612-0328
作者姓名:温博文  董文瀚  解武杰  马 骏
作者单位:空军工程大学 航空航天工程学院,西安 710038
摘    要:随机森林是一种有效的集成学习算法,被广泛应用于模式识别中。为了得到更高的预测精度,需要对参数进行优化。提出了一种基于袋外数据估计的分类误差,利用改进的网格搜索算法对随机森林算法中的决策树数量和候选分裂属性数进行参数优化的随机森林算法。仿真结果表明,利用该方法优化得到的参数都能够使随机森林的分类效果得到一定程度的提高。

关 键 词:随机森林  袋外估计  网格搜索  参数优化  

Parameter optimization method for random forest based on improved grid search algorithm
WEN Bowen,DONG Wenhan,XIE Wujie,MA Jun. Parameter optimization method for random forest based on improved grid search algorithm[J]. Computer Engineering and Applications, 2018, 54(10): 154-157. DOI: 10.3778/j.issn.1002-8331.1612-0328
Authors:WEN Bowen  DONG Wenhan  XIE Wujie  MA Jun
Affiliation:College of Aeronautics and Astronautics Engineering, Air Force Engineering University, Xi’an 710038, China
Abstract:Random forest is an effective ensemble learning method, which is widely used in pattern recognition. In order to get higher accuracy, it is necessary to optimize the parameter of random forest. Based on generalization error of out-of-bag estimates, this paper proposes a parameter optimization method for a random forest with improved grid search. The parameter of the number of decision trees and candidate splitting attributes is optimized to improve accuracy. The simulation results demonstrates that optimized parameter by the method proposed in this paper makes the classification performance of random forest better.
Keywords:random forest  out-of-bag estimates  grid search  parameter optimization  
点击此处可从《计算机工程与应用》浏览原始摘要信息
点击此处可从《计算机工程与应用》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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