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

基于鲸群优化随机森林算法的非平衡数据分类
作者姓名:叶丽珠  郑冬花  刘月红  牛少华
作者单位:1. 广州商学院信息技术与工程学院;2. 马来西亚管理与科学大学研究生院;3. 桂林理工大学信息科学与工程学院;4. 北京理工大学机电学院
基金项目:国家自然科学基金(61961010);;广西省自然科学基金青年基金(2018GXNSFBA050029)资助项目;
摘    要:为了提高非平衡数据分类的准确性,采用随机森林算法用于数据分类,并结合鲸鱼优化算法对随机森林弱分类器权重进行优化求解,以增强随机森林算法对非平衡数据分类的适应性。首先,建立基于随机森林的非平衡数据分类模型。通过随机森林的多个决策树弱分类器进行分类,有效解决样本不均衡导致的分类困难问题。接着,采用鲸群优化算法对弱分类器权重进行优化求解,将分类准确率均值作为鲸群优化适应度函数,以提高弱分类器权重投票对最终分类结果的精度。最后,采用经过鲸群优化得到的随机森林模型进行非平衡数据分类。实验证明,通过合理设置鲸群优化算法参数,可以获得分类准确度更高的随机森林弱分类器权重,相较于常用非平衡数据分类算法,文中算法能够获得更优的分类性能。

关 键 词:非平衡数据分类  随机森林  鲸群优化算法  弱分类器  决策树
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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