首页/ 重庆星环人工智能科技研究院有限公司/ 专利详情

一种机器学习模型的训练方法、计算机设备及存储介质

基本信息

申请号 CN202011591857.7 申请日 -
公开（公告）号 CN112633517A 公开（公告）日 2021-04-09
申请公布号 CN112633517A 申请公布日 2021-04-09
分类号 G06N20/00 分类 计算；推算；计数;
发明人 夏正勋;杨一帆;范豪钧 申请（专利权）人 重庆星环人工智能科技研究院有限公司
代理机构 北京品源专利代理有限公司 代理人 孟金喆
地址 404100 重庆市沙坪坝区凤凰镇皂桷树村临谢家院子组2号1-74室
法律状态 -

摘要

摘要 本发明公开了一种机器学习模型的训练方法、计算机设备及存储介质，该方法包括：从多个样本域中，获取多个第一样本数据和第二样本数据；将各第一样本数据输入至目标模型中，得到多个第一损失函数值，并进行后向计算，得到多个判别网络参数；将第二样本数据输入至加载各判别网络参数的目标模型中，得到多个第二损失函数值；根据各第一损失函数值和各第二损失函数值，计算得到总损失函数值，并根据总损失函数值进行后向计算，返回执行从多个样本域中，获取多个第一样本数据和第二样本数据的操作，直至满足结束训练条件。本发明实施例的技术方案可以实现机器学习模型训练过程的自动化，有效提高机器学习模型在不同场景中的鲁棒性。