easydict模块

EasyDict可以方便地应用 . 来访问dict的值。

麻烦

d = {'foo':3, 'bar':{'x':1, 'y':2}}
print(d['foo'])  # 如何想要访问字典的元素需要这么写
print(d['bar']['y'])  # 如果想要继续访问字典中字典的元素需要使用二维数组
# print(d.foo) 这样写会出错哦!

3
2

简单

from easydict import EasyDict as edict
easy = edict(d = {'foo':3, 'bar':{'x':1, 'y':2}}) # 将普通的字典传入到edict()
print(easy['foo'])  # 这是传统的方法
print(easy.foo)  # 这是我们使用easydict输出二者结果是一样的,但是可以更为方便的使用字典了
print(easy.bar.x)  # 我们也是可以很方便的使用字典中字典的元素了

3
3
1

设置属性

from easydict import EasyDict as edict
d = edict()  # 这个是输出{}
d.foo = 3  # 我们可以直接赋值语句对字典元素进行创建
d.bar = {'prob':'value'}  # 另外我们也可以创建字典中的字典
d.bar.prob = 'newer'  # 另外我们也可以很方便的修改字典中元素的值
print(d)

{'foo': 3, 'bar': {'prob': 'newer'}}

利用 Easydict 建立全局变量

在深度学习中往往利用easydict建立一个全局的变量

from easydict import EasyDict as edict
config = edict() # 创建一个空白的 dict
config.TRAIN = edict() # Train是key,值是{}
config.Test = edict() # Test是key,值是{}
 # config.TRAIN = {} # 这个和上面的那句话是等价的,相当于创建一个字典的key
config.TRAIN.batch_size = 25  # 然后在里面写值,表示Train里面的value也是一个字典
config.TRAIN.early_stopping_num = 10
config.TRAIN.lr = 0.0001
print(config)

{'TRAIN': {'batch_size': 25, 'early_stopping_num': 10, 'lr': 0.0001}, 'Test': {}}

利用 EasyDict 读取配置文件

同样,Easydict 还可以配合 yaml 配置文件一起使用。下面是一个简单的例子。

import yaml
from easydict import EasyDict
def setup_config():
    with open(os.path.join('test_config.yaml')) as f:
        cfg = yaml.safe_load(f) # 读取配置文件
    cfg = EasyDict(cfg) # 存成 Easydict 的格式
    return cfg

参考(感谢)
https://blog.csdn.net/alxe_made/article/details/80507415

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值