From b082eef0988b2b12ccca6fef981c5c1a55f6bbeb Mon Sep 17 00:00:00 2001 From: Jochen Klein Date: Sat, 31 Aug 2024 22:38:09 +0200 Subject: [PATCH] Fix for Run 3 efficiencies --- machine_learning_hep/analysis/analyzer_jets.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/machine_learning_hep/analysis/analyzer_jets.py b/machine_learning_hep/analysis/analyzer_jets.py index 12e48c5621..d2de314109 100644 --- a/machine_learning_hep/analysis/analyzer_jets.py +++ b/machine_learning_hep/analysis/analyzer_jets.py @@ -250,11 +250,10 @@ def calculate_efficiencies(self): self._save_hist(eff, f'eff/h_ptjet-pthf_effnew_{cat}.png') self.h_effnew_ptjet_pthf[cat] = eff - proj_range = (2, get_nbins(h_det[cat], 0)) - eff_avg = project_hist(h_det[cat], [1], {0: proj_range}) + eff_avg = project_hist(h_det[cat], [1], {0: bins_ptjet}) ensure_sumw2(eff_avg) - eff_avg.Divide(project_hist(h_out, [1], {0: proj_range})) - self._save_hist(eff, f'eff/h_pthf_effnew_{cat}.png') + eff_avg.Divide(project_hist(h_out, [1], {0: bins_ptjet})) + self._save_hist(eff_avg, f'eff/h_pthf_effnew_{cat}.png') self.h_effnew_pthf[cat] = eff_avg c = TCanvas() @@ -285,8 +284,7 @@ def _correct_efficiency(self, hist, ipt): if self.cfg('efficiency.correction_method') == 'run3': eff = self.h_effnew_pthf['pr'].GetBinContent(ipt + 1) eff_old = self.hcandeff['pr'].GetBinContent(ipt + 1) - self.logger.info('Using Run 3 efficiency %g instead of %g', - eff, eff_old) + self.logger.info('Using Run 3 efficiency %g instead of %g', eff, eff_old) hist.Scale(1. / eff) elif self.cfg('efficiency.correction_method') == 'run2_2d': self.logger.info('using Run 2 efficiencies per jet pt bin')