Doc improvement: add comment describing what convolve does (openai#264)

RetroAI · Jan 17, 2023 · bb368fe · bb368fe
1 parent 019a593
commit bb368fe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/tutorials/training_agents/blackjack_tutorial.py b/docs/tutorials/training_agents/blackjack_tutorial.py
@@ -299,6 +299,7 @@ def decay_epsilon(self):
 rolling_length = 500
 fig, axs = plt.subplots(ncols=3, figsize=(12, 5))
 axs[0].set_title("Episode rewards")
+# compute and assign a rolling average of the data to provide a smoother graph
 reward_moving_average = (
     np.convolve(
         np.array(env.return_queue).flatten(), np.ones(rolling_length), mode="valid"