機器學習模型訓練方法及裝置

基本信息

申請?zhí)?/td> CN201610716460.3 申請日 -
公開（公告）號 CN107784312A 公開（公告）日 2018-03-09
申請公布號 CN107784312A 申請公布日 2018-03-09
分類號 G06K9/62;G06Q40/02 分類 計算；推算；計數(shù);
發(fā)明人 吳振國 申請（專利權(quán)）人 騰訊征信有限公司
代理機構(gòu) 北京派特恩知識產(chǎn)權(quán)代理有限公司 代理人 張振偉;張穎玲
地址 518057 廣東省深圳市前海深港合作區(qū)前灣一路A棟201室
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種機器學習模型訓練方法及裝置；方法包括：確定訓練集中樣本的散度的差值，和/或所述訓練集中樣本的ROC；基于所述散度的差值和/或所述ROC，構(gòu)造使用所述訓練集訓練的機器學習模型的損失函數(shù)；基于所述訓練集迭代訓練所述機器學習模型；比較相鄰兩次迭代訓練所述機器學習模型后對應(yīng)的損失函數(shù)的相對變化值；當所述損失函數(shù)的相對變化值小于指定值時，停止迭代訓練所述機器學習模型。實施本發(fā)明，能夠克服監(jiān)督學習方式訓練機器學習模型時樣本不均衡帶來的影響，提升機器學習模型的性能。