yaml文件编写规范
更新时间:2023-01-18
yaml文件编写规范
如下所示是进化算法pbt的yaml文件配置示例:
yaml文件中主要分为四大部分:
- 搜索算法参数:
algo:指定选择的搜索策略,下表为各种搜索算法对应的字段。
搜索策略 | 英文字段 |
---|---|
随机搜索 | RANDOM_SEARCH |
贝叶斯搜索 | TPE_SEARCH |
进化算法 | PBT_SEARCH |
进化-CMAES | CMAES_SEARCH |
粒子群算法 | PARTICLE_SEARCH |
params:搜索算法的参数配置,详情见搜索算法简介。
- 评价指标参数:
- name:评价指标的名称。
- goal:优化方向,可选MAXIMIZE/MINIMIZE。
- expected_value:早停标准值,单位为%。
- 搜索参数空间:
- 和手动配置一样,需要填写参数的类型htype以及参数取值范围value,下表为参数类型对应的字段:
参数类型 | 英文字段 |
---|---|
随机整数 | randint |
离散值 | choice |
平均采样 | uniform |
对数平均采样 | loguniform |
离散值的value为所有可能取值的列表,而其余参数类型的value只需要将上下界填入列表即可。
- 其余参数:除上述主要的三部分外,还需填写的内容如下。
- data_sampling_scale:数据采样比例,单位为%。
- max_trial_num:最大搜索次数,随机搜索、贝叶斯搜索的yaml文件中需要配置该参数,而其余基于种群的进化算法不需要,最大搜索次数由种群个体数量与最大轮次的乘积决定。