yaml to dict, adding argparse to dict (easydict, yaml)

 

Simple code to adapt and know how to read yaml and convert it to dict.

And one more thing is add argparse param to dict which is made from yaml.

We use easydict for this.

Refer to below code, then you would understand at a glance.

..

from easydict import EasyDict

import yaml

def load_setting(setting):

with open(setting, 'r') as f:

cfg = yaml.load(f, Loader=yaml.FullLoader)

return EasyDict(cfg)

#----------------------------

cfg = load_setting('test.yaml')

print(cfg)

#{'V1': 'abc', 'V2': {'sub': [1, 2, 3]}}

#----------------------------

#----------------------------

import argparse

paser = argparse.ArgumentParser()

args = paser.parse_args("")

args.batch_size=10

args.epoch=10

#----------------------------

cfg.update(vars(args))

print(cfg, type(cfg))

#{'V1': 'abc', 'V2': {'sub': [1, 2, 3]}, 'batch_size': 10, 'epoch': 10} <class 'easydict.EasyDict'>

#----------------------------

..

Thank you.

www.marearts.com