#include <PhotonXGBoostEstimator.h>

Public Member Functions
float	computeMva (float rawEnergyIn, float r9In, float sigmaIEtaIEtaIn, float etaWidthIn, float phiWidthIn, float s4In, float etaIn, float hOvrEIn, float ecalPFIsoIn) const

	PhotonXGBoostEstimator (const edm::FileInPath &weightsFile, int best_ntree_limit)

Private Attributes
int	best_ntree_limit_ = -1

std::unique_ptr< pat::XGBooster >	booster_

Detailed Description

Definition at line 9 of file PhotonXGBoostEstimator.h.

Constructor & Destructor Documentation

◆ PhotonXGBoostEstimator()

PhotonXGBoostEstimator::PhotonXGBoostEstimator	(	const edm::FileInPath &	weightsFile,
		int	best_ntree_limit
	)

Definition at line 3 of file PhotonXGBoostEstimator.cc.

References best_ntree_limit_, booster_, and pfClustersFromHGC3DClusters_cfi::weightsFile.

                                                                                                      {
   booster_ = std::make_unique<pat::XGBooster>(weightsFile.fullPath());
   booster_->addFeature("rawEnergy");
   booster_->addFeature("r9");
   booster_->addFeature("sigmaIEtaIEta");
   booster_->addFeature("etaWidth");
   booster_->addFeature("phiWidth");
   booster_->addFeature("s4");
   booster_->addFeature("eta");
   booster_->addFeature("hOvrE");
   booster_->addFeature("ecalPFIso");
 
   best_ntree_limit_ = best_ntree_limit;
 }

Member Function Documentation

◆ computeMva()

float PhotonXGBoostEstimator::computeMva	(	float	rawEnergyIn,
		float	r9In,
		float	sigmaIEtaIEtaIn,
		float	etaWidthIn,
		float	phiWidthIn,
		float	s4In,
		float	etaIn,
		float	hOvrEIn,
		float	ecalPFIsoIn
	)		const

Definition at line 18 of file PhotonXGBoostEstimator.cc.

References best_ntree_limit_, and booster_.

                                                                   {
   return booster_->predict(
       {rawEnergyIn, r9In, sigmaIEtaIEtaIn, etaWidthIn, phiWidthIn, s4In, etaIn, hOvrEIn, ecalPFIsoIn},
       best_ntree_limit_);
 }

Member Data Documentation

◆ best_ntree_limit_

int PhotonXGBoostEstimator::best_ntree_limit_ = -1

private

Definition at line 25 of file PhotonXGBoostEstimator.h.

Referenced by computeMva(), and PhotonXGBoostEstimator().

◆ booster_

std::unique_ptr<pat::XGBooster> PhotonXGBoostEstimator::booster_