pomdp

2021-06-01 12:39:33 +02:00
parent 403d38dc24
commit 55b409c72f
6 changed files with 82 additions and 83 deletions
--- a/environments/logging/plotting.py
+++ b/environments/logging/plotting.py
@@ -25,8 +25,9 @@ def plot(filepath, ext='png', **kwargs):

    plt.tight_layout()
    figure = plt.gcf()
-    plt.show()
    figure.savefig(str(filepath), format=ext)
+    plt.show()
+    plt.clf()


 def prepare_plot(filepath, results_df, ext='png'):