fix evaluate_agent (issue #37)

This commit is contained in:
unfinity
2022-05-24 13:41:55 +03:00
parent f74adfb0a2
commit 29e44a2e45

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",