Fix bug with agent evaluation for very long episodes

fix bug in evaluate_agent (issue #37)
This commit is contained in:
Omar Sanseviero
2022-05-24 14:08:00 +02:00
committed by GitHub

View File

@@ -892,9 +892,9 @@
" total_rewards_ep += reward\n",
" \n",
" if done:\n",
" episode_rewards.append(total_rewards_ep)\n",
" break\n",
" state = new_state\n",
" episode_rewards.append(total_rewards_ep)\n",
" mean_reward = np.mean(episode_rewards)\n",
" std_reward = np.std(episode_rewards)\n",
"\n",