参数配置文件:yaml的基础使用
在训练深度学习网络过程中,由于涉及到很多参数,通常会使用专门的参数配置文件,将需要调整的参数全部写在一个文件中以便管理。这样的参数配置文件通常有json、yaml、yacs等等格式。下面则以yaml为例进行介绍。
一、yaml基本介绍
YAML 是 “YAML Ain’t a Markup Language”(YAML 不是一种标记语言)的递归缩写。在开发的这种语言时,YAML 的意思其实是:“Yet Another Markup Language”(仍是一种标记语言)。
1、基本语法
- 大小写敏感
- 使用缩进表示层级关系
- 缩进不允许使用tab,只允许空格
- 缩进的空格数不重要,只要相同层级的元素左对齐即可
- '#'表示注释
- 文件中的字符串不需要使用引号标注,但若字符串包含有特殊字符则需用引号标注
2、数据类型
YAML 支持以下几种数据类型:
- 对象:键值对的集合,映射/ 哈希/ 字典
键值对用冒号 “:” 结构表示,冒号与值之间需用空格分隔
- 数组:一组按次序排列的值,序列 / 列表
数组前加有 “-” 符号,符号与值之间需用空格分隔
- 纯量(scalars):单个的、不可再分的值。
如字符串、bool值、整数、浮点数、时间、日期、null等(None值可用null可"~"表示)
二、python读取yaml参数配置文件
1、前提
前提是虚拟环境中已经安装了yaml库。并且在对应的py脚本中导入yaml库。
2、读取文件
python通过open方式读取文件数据,再通过load函数将数据转化为列表或字典。
示例代码如下:
import yaml
with open('yaml_test.yml', 'r', encoding='utf-8') as y:
cfg = yaml.load(y)
print(cfg)
3、示例
(1) 数组
# 数组
- 1
- 13
- 34
python解析输出结果:
[1, 13, 34]
(2) 字典
# 键值对(对象)
age: 18
name: python
python解析输出结果:
{
'age': 18,