From 9db4ffce315c45569dad19f74074146dc2cae424 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Tue, 19 Jun 2018 11:41:32 +0200
Subject: [PATCH] Close figures after plotting
---
toolkit.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index 8cb4d72..51d7aae 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -859,7 +859,7 @@ class ClassificationProject(object):
if not os.path.exists(plot_dir):
os.mkdir(plot_dir)
fig.savefig(os.path.join(plot_dir, "var_{}.pdf".format(var_index)))
- plt.clf()
+ plt.close(fig)
def plot_weights(self):
@@ -868,9 +868,11 @@ class ClassificationProject(object):
sig = self.w_train[self.y_train == 1]
ax.hist(bkg, bins=100, color="b", alpha=0.5)
fig.savefig(os.path.join(self.project_dir, "eventweights_bkg.pdf"))
+ plt.close(fig)
fig, ax = plt.subplots()
ax.hist(sig, bins=100, color="r", alpha=0.5)
fig.savefig(os.path.join(self.project_dir, "eventweights_sig.pdf"))
+ plt.close(fig)
def plot_ROC(self):
@@ -914,6 +916,7 @@ class ClassificationProject(object):
ax.set_xlabel("NN output")
fig.legend(loc='upper center', framealpha=0.5)
fig.savefig(os.path.join(self.project_dir, "scores.pdf"))
+ plt.close(fig)
def plot_significance(self, lumifactor=1., significanceFunction=None, plot_opts=dict(bins=50, range=(0, 1))):
@@ -964,6 +967,7 @@ class ClassificationProject(object):
ax.set_ylabel("Significance")
ax.legend(loc='lower center', framealpha=0.5)
fig.savefig(os.path.join(self.project_dir, "significances.pdf"))
+ plt.close(fig)
@property
--
GitLab