File indexing completed on 2024-06-18 02:20:36
0001 #ifndef ReciEgamma_PhotonIdentification_PhotonXGBoostEstimator_h
0002 #define ReciEgamma_PhotonIdentification_PhotonXGBoostEstimator_h
0003
0004 #include "FWCore/ParameterSet/interface/FileInPath.h"
0005 #include "PhysicsTools/XGBoost/interface/XGBooster.h"
0006
0007 #include <memory>
0008
0009 class PhotonXGBoostEstimator {
0010 public:
0011 PhotonXGBoostEstimator(const edm::FileInPath& weightsFile, int best_ntree_limit);
0012
0013 float computeMva(float rawEnergyIn,
0014 float r9In,
0015 float sigmaIEtaIEtaIn,
0016 float etaWidthIn,
0017 float phiWidthIn,
0018 float s4In,
0019 float etaIn,
0020 float hOvrEIn,
0021 float ecalPFIsoIn) const;
0022
0023 private:
0024 std::unique_ptr<pat::XGBooster> booster_;
0025 int best_ntree_limit_ = -1;
0026 };
0027
0028 #endif