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}