gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-01 22:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Luke Meyers
2023-01-17 10:06:11 -05:00
committed by GitHub
parent 019a593117
commit bb368fe75f
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/tutorials/training_agents/blackjack_tutorial.py
View File

@@ -299,6 +299,7 @@ for episode in tqdm(range(n_episodes)):
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"