Загрузка гиперпараметров из YML в Python - PullRequest
Новая
       3

Загрузка гиперпараметров из YML в Python

0 голосов
/ 12 января 2020

Как мне импортировать и использовать настройки в файле YML в моем коде Python? 100

пример файла YML. config.yml 

HalfCheetah-v2:
  env_wrapper: utils.wrappers.TimeFeatureWrapper
  n_timesteps: !!float 1e6
  policy: 'MlpPolicy'
  gamma: 0.99
  buffer_size: 1000000
  noise_type: 'normal'
  noise_std: 0.1
  learning_starts: 10000
  batch_size: 100
  learning_rate: !!float 1e-3
  train_freq: 1000
  gradient_steps: 1000
  policy_kwargs: "dict(layers=[400, 300])"

пример файла Python, train.py 

import gym
import config..? 

env_id = 'HalfCheetah-v2'
env = DummyVecEnv([lambda: gym.make(env_id)])

# Settings should go here TD3(<>)   
model = TD3()
...