def retro_analyze_sessions(predir):
'''Retro analyze all sessions'''
logger.info('Running retro_analyze_sessions')
session_spec_paths = glob(f'{predir}/*_s*_spec.json')
util.parallelize(_retro_analyze_session,
[(p, ) for p in session_spec_paths],
num_cpus=util.NUM_CPUS)
def retro_analyze_trials(predir):
'''Retro analyze all trials'''
logger.info('Running retro_analyze_trials')
session_spec_paths = glob(f'{predir}/*_s*_spec.json')
# remove session spec paths
trial_spec_paths = ps.difference(glob(f'{predir}/*_t*_spec.json'), session_spec_paths)
util.parallelize(_retro_analyze_trial, [(p,) for p in trial_spec_paths], num_cpus=util.NUM_CPUS)
def gen_random_baseline(env_name, num_eval=NUM_EVAL):
'''Generate the random baseline for an environment by averaging over num_eval episodes'''
returns = util.parallelize(gen_random_return,
[(env_name, i) for i in range(num_eval)])
mean_rand_ret = np.mean(returns)
std_rand_ret = np.std(returns)
return {'mean': mean_rand_ret, 'std': std_rand_ret}
