资讯 社区 文档
技术能力
语音技术
文字识别
人脸与人体
图像技术
语言与知识
视频技术

yaml文件编写规范

yaml文件编写规范

如下所示是进化算法pbt的yaml文件配置示例: image.png

yaml文件中主要分为四大部分:

  1. 搜索算法参数:

algo:指定选择的搜索策略,下表为各种搜索算法对应的字段。

搜索策略 英文字段
随机搜索 RANDOM_SEARCH
贝叶斯搜索 TPE_SEARCH
进化算法 PBT_SEARCH
进化-CMAES CMAES_SEARCH
粒子群算法 PARTICLE_SEARCH

params:搜索算法的参数配置,详情见搜索算法简介。

  1. 评价指标参数:
  • name:评价指标的名称。
  • goal:优化方向,可选MAXIMIZE/MINIMIZE。
  • expected_value:早停标准值,单位为%。
  1. 搜索参数空间:
  • 和手动配置一样,需要填写参数的类型htype以及参数取值范围value,下表为参数类型对应的字段:
参数类型 英文字段
随机整数 randint
离散值 choice
平均采样 uniform
对数平均采样 loguniform

离散值的value为所有可能取值的列表,而其余参数类型的value只需要将上下界填入列表即可。

  1. 其余参数:除上述主要的三部分外,还需填写的内容如下。
  • data_sampling_scale:数据采样比例,单位为%。
  • max_trial_num:最大搜索次数,随机搜索、贝叶斯搜索的yaml文件中需要配置该参数,而其余基于种群的进化算法不需要,最大搜索次数由种群个体数量与最大轮次的乘积决定。
上一篇
自动搜索作业代码编写规范
下一篇
自动搜索作业代码示例