[Python] yaml file 읽기(parameter 지정)

0. package install

먼저, python yaml 패키지가 설치되어야 합니다.

!pip install PyYAML

1. yaml import 후 읽기

그 후, yaml를 import 한 후 읽습니다.

import yaml

with open('./parameter.yaml') as f:
	parameters = yaml.load(f, Loader = yaml.FullLoader)

2. parameter 지정

yaml file은 key와 value 값으로 구성되어 있기 때문에 key 값으로 value를 불러옵니다.

learning_rate = parameters['learning_rate']
epoch = parameters['epoch']