PhotonXGBoostEstimator.cc

Go to the documentation of this file.

#include "RecoEgamma/PhotonIdentification/interface/PhotonXGBoostEstimator.h"

PhotonXGBoostEstimator::PhotonXGBoostEstimator(const edm::FileInPath& weightsFile, int best_ntree_limit) {
  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;
}

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