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