← 返回列表

代码优化方法、装置、设备及介质

申请号: CN201810940377.3
申请人: 第四范式(北京)技术有限公司
申请日期: 2018年8月17日

摘要文本

本发明提出了一种机器学习模型中超参数的调优方法、装置、设备及介质。获取用于训练机器学习模型的模板代码,其中,模板代码中标识出了一种或多种超参数中每种超参数各自的取值集合;对模板代码进行解析,以识别一种或多种超参数各自的取值集合;基于一种或多种超参数各自的取值集合,生成一种或多种超参数的至少部分取值组合;获取按照至少部分取值组合中的每种取值组合来训练机器学习模型所得到的关于模型效果的评测结果;以及基于评测结果,确定机器学习模型的一种或多种超参数的优化取值。由此,可以降低人工调参的消耗,有效地提高机器学习(如深度学习)的开发效率。

专利详细信息

项目 内容
专利名称 代码优化方法、装置、设备及介质
专利类型 发明授权
申请号 CN201810940377.3
申请日 2018年8月17日
公告号 CN109242105B
公开日 2024年3月15日
IPC主分类号 G06N20/00
权利人 第四范式(北京)技术有限公司
发明人 裴炜欣; 赵汉光; 王珵; 戴文渊
地址 北京市海淀区上地东路35号颐泉汇大厦写字楼A座610室

专利主权项内容

1.一种代码优化方法,包括:获取用于训练机器学习模型的模板代码,其中,所述模板代码中标识出了一种或多种超参数中每种超参数各自的取值集合,所述取值集合是以不同于代码的形式标识的;对所述模板代码进行解析,以识别所述一种或多种超参数各自的取值集合;基于所述一种或多种超参数各自的取值集合,生成所述一种或多种超参数的至少部分取值组合;获取按照所述至少部分取值组合中的每种取值组合来训练所述机器学习模型所得到的关于模型效果的评测结果,包括:基于对所述模板代码进行解析的结果,将所述模板代码中各个取值集合替换为取值组合中相应的超参数的取值,对替换后得到的模板代码进行编译,生成与所述至少部分取值组合中的每种取值组合分别对应的多份可执行代码;通过运行所述多份可执行代码来训练所述机器学习模型,并获得相应的评测结果;以及基于所述评测结果,确定所述机器学习模型的所述一种或多种超参数的优化取值。