Add acceleration.

MatthewGerber · Jul 26, 2024 · 70a9cbc · 70a9cbc
1 parent 993db67
commit 70a9cbc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/rlai/policy_gradient/monte_carlo/reinforce.py b/src/rlai/policy_gradient/monte_carlo/reinforce.py
@@ -360,6 +360,7 @@ def improve(
                         label=plot_label,
                         **environment.plot_label_data_kwargs[plot_label][1]
                     )
+                plt.legend()
                 plt.tight_layout()
 
                 if pdf is None: