具有强化学习策略的决策树算法 |
| |
作者单位: | 辽宁科技大学 电子与信息工程学院,辽宁 鞍山 114000;烟台汽车工程职业学院 汽车工程系,山东 烟台 265500 |
| |
基金项目: | 国家自然科学基金;辽宁省自然科学基金;辽宁省高等学校基本科研资助项目 |
| |
摘 要: | 传统决策树在对不平衡数据进行分类时,提高正类的权重和舍弃部分负类的信息,造成负类的预测精度较低。文章引入强化学习思想,提出一种基于马尔可夫决策过程的改进决策树方法。根据马尔可夫决策过程、当前分裂特征的标准化互信息和马修斯相关系数作为信息增益率的奖励或者惩罚,形成新的特征选择标准。实验结果表明,与其他传统方法相比,改进的马尔可夫决策树对非平衡数据整体的预测精度及负类预测精度均有提高。
|
关 键 词: | 决策树 不平衡数据 强化学习 标准化互信息 马修斯相关系数 |
本文献已被 CNKI 万方数据 等数据库收录! |
|