使用configparser
模块读取模型参数,设置config.ini文件内容如下,[train]
和[savepath]
分别为两个session
[train]
lr = 1e-4
epochs = 200
dropout = 0.5
batch_size = 4
[savepath]
path = ../data/
基本操作如下
- 读取配置文件
import configparser
config = configparser.RawConfigParser()
config.read("config.ini")
- 获取文件的所有section
secs = config.sections()
print(secs)
>>> ['train', 'savepath']
- 获取指定section下的所有参数key
options = config.options("train")
print(options)
>>> ['lr', 'epochs', 'dropout', 'batch_size']
- 获取指定section中指定key的value(最常用),可以使用get获取字符串,getint获取整型,getfloat获取浮点型,getboolean获取布尔值
learning_rate = config.getfloat("train", "lr")
print(learning_rate , type(learning_rate))
>>> 0.0001 <class 'float'>
path = config.get('filepath', 'path')
print(path, type(path))
>>> ../data/ <class 'str'>