higgs_dna.tools package

Submodules

higgs_dna.tools.chained_quantile module

class higgs_dna.tools.chained_quantile.ChainedQuantileRegression(corrections_summary: str, SS_variables: List[str], correctShowerShapes: bool = True, correctIsolations: bool = True, correctPreshower: bool = False)[source]

Bases: object

apply(photons: Array, events: Array) → Array[source]

apply_charged_isolation(photons: Array, rho: Array, isEB: ndarray, isEE: ndarray) → Array[source]

apply_photon_isolation(photons: Array, rho: Array, isEB: ndarray, isEE: ndarray) → Array[source]

apply_shower_shapes(photons: Array, rho: Array, isEB: ndarray, isEE: ndarray) → Array[source]

higgs_dna.tools.chained_quantile.create_evaluator(weights: str, scale: float | None = None, center: float | None = None, variables: List[str] | None = None, **kwargs: Dict[Any, Any]) → wrapped_xgb[source]

class higgs_dna.tools.chained_quantile.wrapped_xgb(model: Booster, scale: float | None, center: float | None, variables: List[str] | None)[source]

Bases: object

property center: float | None

property scale: float | None

property variables: List[str] | None

higgs_dna.tools.diphoton_mva module

higgs_dna.tools.diphoton_mva.calculate_diphoton_mva(mva: Tuple[Booster | None, List[str]], diphotons: Array, events: Array) → Array[source]: Calculate DiphotonID bdt scores for diphoton.

higgs_dna.tools.diphoton_mva.calculate_retrained_diphoton_mva(self, mva: Tuple[Booster | None, List[str]], diphotons: Array, events: Array) → Array[source]: Calculate DiphotonID bdt scores for diphoton using retrained bdt that avoids flashgg inputs. To calculate the score I need some extra variables.

higgs_dna.tools.photonid_mva module

higgs_dna.tools.photonid_mva.calculate_photonid_mva(mva: Tuple[Booster | None, List[str]], photon: Array) → Array[source]

Recompute PhotonIDMVA on-the-fly. This step is necessary considering that the inputs have to be corrected with the QRC process. Following is the list of features (barrel has 12, endcap two more): EB:

events.Photon.energyRaw events.Photon.r9 events.Photon.sieie events.Photon.etaWidth events.Photon.phiWidth events.Photon.sieip events.Photon.s4 events.Photon.pfPhoIso03 events.Photon.pfChargedIsoPFPV events.Photon.pfChargedIsoWorstVtx events.Photon.eta events.fixedGridRhoAll

EE: EB +: events.Photon.esEffSigmaRR events.Photon.esEnergyOverRawE

higgs_dna.tools.photonid_mva.load_photonid_mva(fname: str) → Booster | None[source]

higgs_dna.tools.xgb_loader module

higgs_dna.tools.xgb_loader.load_bdt(fname: str) → Booster | None[source]