mirror of
https://github.com/huggingface/deep-rl-class.git
synced 2026-04-05 11:38:43 +08:00
Fix bug with agent evaluation for very long episodes
fix bug in evaluate_agent (issue #37)
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user