From e61e3bf16cb50008a2374f2dc0b92f26ec918255 Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Thu, 23 Aug 2018 10:35:58 +0200
Subject: [PATCH] fixing recreation of w_train_tot when reloading
---
toolkit.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 3498ab3..609a1f6 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -379,6 +379,8 @@ class ClassificationProject(object):
def _load_data(self):
+ self._w_train_tot = None
+
try:
# if those don't exist, we need to load them from ROOT trees first
@@ -1113,11 +1115,11 @@ class ClassificationProject(object):
sig = self.w_train_tot[self.y_train == 1]
ax.hist(bkg, bins=bins, range=range, color="b", alpha=0.5)
ax.set_yscale("log")
- return save_show(plt, fig, os.path.join(self.project_dir, "eventweights_bkg.pdf"))
+ save_show(plt, fig, os.path.join(self.project_dir, "eventweights_bkg.pdf"))
fig, ax = plt.subplots()
ax.hist(sig, bins=bins, range=range, color="r", alpha=0.5)
ax.set_yscale("log")
- return save_show(plt, fig, os.path.join(self.project_dir, "eventweights_sig.pdf"))
+ save_show(plt, fig, os.path.join(self.project_dir, "eventweights_sig.pdf"))
def plot_ROC(self, xlim=(0,1), ylim=(0,1)):
@@ -1583,6 +1585,7 @@ class ClassificationProjectDataFrame(ClassificationProject):
self._y_test = None
self._w_train = None
self._w_test = None
+ self._w_train_tot = None
if not self.data_transformed:
self._transform_data()
--
GitLab