Timestep based test scheme

This commit is contained in:
Shangtong Zhang
2018-04-12 22:23:21 -06:00
parent de478faf9b
commit 08d45b9284
6 changed files with 46 additions and 19 deletions
-9
View File
@@ -39,15 +39,6 @@ def run_episodes(agent):
if config.max_steps and agent.total_steps > config.max_steps:
break
if config.test_interval and ep % config.test_interval == 0:
test_rewards = []
for _ in range(config.test_repetitions):
test_rewards.append(agent.episode(True)[0])
avg_reward = np.mean(test_rewards)
avg_test_rewards.append(avg_reward)
config.logger.info('Averaged test reward %f(%f)' % (
avg_reward, np.std(test_rewards) / np.sqrt(config.test_repetitions)))
agent.close()
return steps, rewards, avg_test_rewards