基于二分类业务的数据特征选择方法、装置及计算机设备

摘要

本发明实施例所提供的基于二分类业务的数据特征选择方法、装置及计算机设备，采用提取得到的初始特征组合对第一树模型进行训练并获取初始增益贡献值并从n个初始数据特征筛选出m个第一数据特征，确定第一数据特征对应的相关性系数以及特征iv值，基于相关性系数和特征iv值从m个第一数据特征中筛选出j个第二数据特征并对第二树模型进行训练以获取当前增益贡献值，根据当前增益贡献值和绘制的iv图从j个第二数据特征中筛选出k个第三数据特征。如此，能够将数据特征的增益贡献、特征间相关性和特征iv值考虑在内，从而快速高效地选择出数量少且具有不可替代性的数据特征，这样可以减少对树模型进行建模和训练的耗时并降低计算机的处理负荷。