From b40483204cde6e3cf20eb223960967cad9889343 Mon Sep 17 00:00:00 2001
From: "Alwin.Wester"
<Alwin.Wester@gar-ws-etp10.garching.physik.uni-muenchen.de>
Date: Mon, 30 Apr 2018 18:57:24 +0200
Subject: [PATCH] debug plot for event weights
---
toolkit.py | 20 +++++++++++++++++---
1 file changed, 17 insertions(+), 3 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 9904ec7..8f10b92 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -351,6 +351,7 @@ class KerasROOTClassification(object):
sumw_bkg = np.sum(self.w_train[self.y_train == 0])
sumw_sig = np.sum(self.w_train[self.y_train == 1])
self._class_weight = [(sumw_sig+sumw_bkg)/(2*sumw_bkg), (sumw_sig+sumw_bkg)/(2*sumw_sig)]
+ logger.debug("Calculated class_weight: {}".format(self._class_weight))
return self._class_weight
@@ -489,6 +490,18 @@ class KerasROOTClassification(object):
fig.savefig(os.path.join(plot_dir, "var_{}.pdf".format(var_index)))
plt.clf()
+
+ def plot_weights(self):
+ fig, ax = plt.subplots()
+ bkg = self.w_train[self.y_train == 0]
+ 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"))
+ 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"))
+
+
def plot_ROC(self):
logger.info("Plot ROC curve")
@@ -580,7 +593,8 @@ if __name__ == "__main__":
step_bkg = 100)
c.load()
- #c.train(epochs=20)
+ c.train(epochs=20)
c.plot_ROC()
- # c.plot_loss()
- # c.plot_accuracy()
+ c.plot_loss()
+ c.plot_accuracy()
+ c.plot_weights()
--
GitLab