From 2434384af3db2f99939f00f93f8788c117e6d361 Mon Sep 17 00:00:00 2001
From: Thomas Weber <Thomas.Weber@physik.uni-muenchen.de>
Date: Fri, 8 Jun 2018 13:41:18 +0200
Subject: [PATCH] Write max significance into significance plot
---
toolkit.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 7e1954a..9605aa4 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -878,6 +878,8 @@ class ClassificationProject(object):
def plot_significance(self, lumifactor=1., significanceFunction=None):
+ logger.info("Plot significances")
+
plot_opts = dict(bins=50, range=(0, 1))
centers_sig_train, hist_sig_train, rel_errors_sig_train = self.get_bin_centered_hist(self.scores_train[self.y_train==1].reshape(-1), weights=self.w_train[self.y_train==1], **plot_opts)
centers_bkg_train, hist_bkg_train, rel_errors_bkg_train = self.get_bin_centered_hist(self.scores_train[self.y_train==0].reshape(-1), weights=self.w_train[self.y_train==0], **plot_opts)
@@ -911,8 +913,8 @@ class ClassificationProject(object):
fig, ax = plt.subplots()
width = centers_sig_train[1]-centers_sig_train[0]
- ax.plot(centers_bkg_train, significances_train, label="train")
- ax.plot(centers_bkg_test, significances_test, label="test")
+ ax.plot(centers_bkg_train, significances_train, label="train, Z_max={}".format(np.amax(significances_train)))
+ ax.plot(centers_bkg_test, significances_test, label="test, Z_max={}".format(np.amax(significances_test)))
ax.set_xlabel("Cut on NN score")
ax.set_ylabel("Significance")
ax.legend(loc='lower center', framealpha=0.5)
@@ -993,6 +995,7 @@ class ClassificationProject(object):
self.plot_loss()
self.plot_score()
self.plot_weights()
+ self.plot_significance()
def create_getter(dataset_name):
--
GitLab