Fixing error bars in score plot

toolkit.py

@@ -762,6 +762,7 @@ class ClassificationProject(object):
 
     @staticmethod
     def get_bin_centered_hist(x, scale_factor=None, **np_kwargs):
+        "Return bin centers, histogram and relative (!) errors"
         hist, bins = np.histogram(x, **np_kwargs)
         centers = (bins[:-1] + bins[1:]) / 2
         if "weights" in np_kwargs:
@@ -859,8 +860,10 @@ class ClassificationProject(object):
         plot_opts = dict(bins=50, range=(0, 1))
         centers_sig_train, hist_sig_train, _ = self.get_bin_centered_hist(self.scores_train[self.y_train==1].reshape(-1), density=True, weights=self.w_train[self.y_train==1], **plot_opts)
         centers_bkg_train, hist_bkg_train, _ = self.get_bin_centered_hist(self.scores_train[self.y_train==0].reshape(-1), density=True, weights=self.w_train[self.y_train==0], **plot_opts)
-        centers_sig_test, hist_sig_test, errors_sig_test = self.get_bin_centered_hist(self.scores_test[self.y_test==1].reshape(-1), density=True, weights=self.w_test[self.y_test==1], **plot_opts)
-        centers_bkg_test, hist_bkg_test, errors_bkg_test = self.get_bin_centered_hist(self.scores_test[self.y_test==0].reshape(-1), density=True, weights=self.w_test[self.y_test==0], **plot_opts)
+        centers_sig_test, hist_sig_test, rel_errors_sig_test = self.get_bin_centered_hist(self.scores_test[self.y_test==1].reshape(-1), density=True, weights=self.w_test[self.y_test==1], **plot_opts)
+        centers_bkg_test, hist_bkg_test, rel_errors_bkg_test = self.get_bin_centered_hist(self.scores_test[self.y_test==0].reshape(-1), density=True, weights=self.w_test[self.y_test==0], **plot_opts)
+        errors_sig_test = hist_sig_test*rel_errors_sig_test
+        errors_bkg_test = hist_bkg_test*rel_errors_bkg_test
         fig, ax = plt.subplots()
         width = centers_sig_train[1]-centers_sig_train[0]
         ax.bar(centers_bkg_train, hist_bkg_train, color="b", alpha=0.5, width=width, label="background train")