From 818ef9631d8832ff726619ea5b1bf4df819a3e13 Mon Sep 17 00:00:00 2001 From: Andrei Cuceu Date: Sat, 11 Nov 2023 21:48:32 -0500 Subject: [PATCH] Test fast bb --- vega/minimizer.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/vega/minimizer.py b/vega/minimizer.py index 2443045..9a3992f 100644 --- a/vega/minimizer.py +++ b/vega/minimizer.py @@ -88,9 +88,19 @@ def minimize(self, params=None): for param, value in minuit_biases.values.to_dict().items(): self.params_init[param] = value - # If we have broadband polynomials, we first maximize one correlation at a time + print(self.params_init) + # If we have broadband polynomials we first maximize all other parameters, + # and then the broadband parameters one correlation at a time bb_params = [par for par in self._names if 'BB-' in par] if bool(len(bb_params)): + cf_bb_params = [par for par in self._names if 'BB-' not in par] + minuit_bb = self.run_iminuit(cf_bb_params) + + for param, value in minuit_bb.values.to_dict().items(): + self.params_init[param] = value + + print(self.params_init) + for cf_name in self._cf_names: cf_bb_params = [par for par in bb_params if cf_name in par] minuit_bb = self.run_iminuit(cf_bb_params) @@ -98,6 +108,8 @@ def minimize(self, params=None): for param, value in minuit_bb.values.to_dict().items(): self.params_init[param] = value + print(self.params_init) + # Do the actual minimization self._minuit = self.run_iminuit(self._names)