机器学习模型训练方法及装置

基本信息

申请号 CN201610716460.3 申请日 -
公开(公告)号 CN107784312A 公开(公告)日 2018-03-09
申请公布号 CN107784312A 申请公布日 2018-03-09
分类号 G06K9/62;G06Q40/02 分类 计算;推算;计数;
发明人 吴振国 申请(专利权)人 腾讯征信有限公司
代理机构 北京派特恩知识产权代理有限公司 代理人 张振伟;张颖玲
地址 518057 广东省深圳市前海深港合作区前湾一路A栋201室
法律状态 -

摘要

摘要 本发明公开了一种机器学习模型训练方法及装置;方法包括:确定训练集中样本的散度的差值,和/或所述训练集中样本的ROC;基于所述散度的差值和/或所述ROC,构造使用所述训练集训练的机器学习模型的损失函数;基于所述训练集迭代训练所述机器学习模型;比较相邻两次迭代训练所述机器学习模型后对应的损失函数的相对变化值;当所述损失函数的相对变化值小于指定值时,停止迭代训练所述机器学习模型。实施本发明,能够克服监督学习方式训练机器学习模型时样本不均衡带来的影响,提升机器学习模型的性能。