PhotonXGBoostEstimator Class Reference

#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)
	~PhotonXGBoostEstimator ()

Private Attributes
int	best_ntree_limit_ = -1
BoosterHandle	booster_
std::string	config_

Detailed Description

Definition at line 7 of file PhotonXGBoostEstimator.h.

Constructor & Destructor Documentation

PhotonXGBoostEstimator()

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

Definition at line 4 of file PhotonXGBoostEstimator.cc.

References best_ntree_limit_, booster_, DiMuonV_cfg::config, config_, NULL, and pfClustersFromHGC3DClusters_cfi::weightsFile.

                                                                                                     {
  XGBoosterCreate(NULL, 0, &booster_);
  // Set number of threads to 1, to avoid spawning hundreds of OpenMP threads
  // See https://github.com/cms-sw/cmssw/issues/44923 for details
  XGBoosterSetParam(booster_, "nthread", "1");
  XGBoosterLoadModel(booster_, weightsFile.fullPath().c_str());
  best_ntree_limit_ = best_ntree_limit;

  std::stringstream config;
  config << "{\"training\": false, \"type\": 0, \"iteration_begin\": 0, \"iteration_end\": " << best_ntree_limit_
         << ", \"strict_shape\": false}";
  config_ = config.str();
}

~PhotonXGBoostEstimator()

PhotonXGBoostEstimator::~PhotonXGBoostEstimator

(

)

Definition at line 18 of file PhotonXGBoostEstimator.cc.

References booster_.

{ XGBoosterFree(booster_); }

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 34 of file PhotonXGBoostEstimator.cc.

References booster_, config_, PVValHelper::eta, electrons_cff::etaWidth, f, NULL, electrons_cff::phiWidth, electrons_cff::r9, electrons_cff::rawEnergy, runTheMatrix::ret, photons_cff::s4, and trigObjTnPSource_cfi::var.

                                                                  {
  float var[9];
  var[rawEnergy] = rawEnergyIn;
  var[r9] = r9In;
  var[sigmaIEtaIEta] = sigmaIEtaIEtaIn;
  var[etaWidth] = etaWidthIn;
  var[phiWidth] = phiWidthIn;
  var[s4] = s4In;
  var[eta] = etaIn;
  var[hOvrE] = hOvrEIn;
  var[ecalPFIso] = ecalPFIsoIn;

  DMatrixHandle dmat;
  XGDMatrixCreateFromMat(var, 1, 9, -999.9f, &dmat);
  uint64_t const* out_shape;
  uint64_t out_dim;
  const float* out_result = NULL;
  XGBoosterPredictFromDMatrix(booster_, dmat, config_.c_str(), &out_shape, &out_dim, &out_result);
  float ret = out_result[0];
  XGDMatrixFree(dmat);
  return ret;
}

Member Data Documentation

best_ntree_limit_

int PhotonXGBoostEstimator::best_ntree_limit_ = -1

private

Definition at line 24 of file PhotonXGBoostEstimator.h.

Referenced by PhotonXGBoostEstimator().

booster_

BoosterHandle PhotonXGBoostEstimator::booster_

private

Definition at line 23 of file PhotonXGBoostEstimator.h.

Referenced by computeMva(), PhotonXGBoostEstimator(), and ~PhotonXGBoostEstimator().

config_

std::string PhotonXGBoostEstimator::config_

private

Definition at line 25 of file PhotonXGBoostEstimator.h.

Referenced by computeMva(), and PhotonXGBoostEstimator().