22 |
|
#include "MitCommon/MathTools/interface/MathUtils.h" |
23 |
|
|
24 |
|
#include "MitPhysics/Utils/interface/RecoilTools.h" |
25 |
+ |
#include "MitPhysics/Utils/interface/GBRForest.h" |
26 |
|
|
27 |
|
class TRandom3; |
28 |
|
|
35 |
|
kBaseline = 0 |
36 |
|
}; |
37 |
|
|
38 |
< |
void setVariables(TMVA::Reader *iReader,bool iScale); |
39 |
< |
void Initialize( TString iU1MethodName="U1MVA", |
40 |
< |
TString iPhiMethodName="PhiMVA", |
41 |
< |
TString iJetMVAFile="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_data_clean2_njet.weights.xml", |
42 |
< |
TString iU1Weights="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilRegress_clean2_njet.weights.xml", |
43 |
< |
TString iPhiWeights="$CMSSW_BASE/src/MitPhysics/data/mva_JetID.weights.xml", |
44 |
< |
MVAType iType=kBaseline); |
45 |
< |
|
38 |
> |
//void setVariables(TMVA::Reader *iReader,bool iScale); |
39 |
> |
//void Initialize( TString iU1MethodName="U1MVA", |
40 |
> |
// TString iPhiMethodName="PhiMVA", |
41 |
> |
// TString iJetMVAFile="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml", |
42 |
> |
// TString iU1Weights="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilRegress_baseline.weights.xml", |
43 |
> |
// TString iPhiWeights="$CMSSW_BASE/src/MitPhysics/data/mva_JetID.weights.xml", |
44 |
> |
// MVAType iType=kBaseline); |
45 |
> |
void Initialize( |
46 |
> |
TString iJetMVAFile="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml", |
47 |
> |
TString iU1Weights ="$CMSSW_BASE/src/MitPhysics/data/gbrmet.root", |
48 |
> |
TString iPhiWeights="$CMSSW_BASE/src/MitPhysics/data/gbrmetphi.root", |
49 |
> |
MVAType iType=kBaseline); |
50 |
> |
|
51 |
|
Bool_t IsInitialized() const { return fIsInitialized; } |
52 |
+ |
Double_t evaluatePhi(); |
53 |
+ |
Double_t evaluateU1(); |
54 |
|
Double_t MVAValue( bool iPhi, |
55 |
|
Float_t iPFSumEt, |
56 |
|
Float_t iU , |
86 |
|
int iNPV, |
87 |
|
Bool_t printDebug=false); |
88 |
|
|
89 |
+ |
Met GetMet( Bool_t iPhi,Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis, |
90 |
+ |
const PFMet *iMet , |
91 |
+ |
const PFCandidateCol *iCands,const Vertex *iVertex, |
92 |
+ |
const PFJetCol *iJets , |
93 |
+ |
int iNPV, |
94 |
+ |
Bool_t printDebug=false); |
95 |
+ |
|
96 |
|
Met GetMet( Bool_t iPhi, |
97 |
|
Float_t iPt1,Float_t iPhi1,Float_t iEta1, |
98 |
|
Float_t iPt2,Float_t iPhi2,Float_t iEta2, |
103 |
|
const PileupEnergyDensityCol *iPUEnergyDensity, |
104 |
|
int iNPV, |
105 |
|
Bool_t printDebug); |
106 |
+ |
|
107 |
+ |
Met GetMet( Bool_t iPhi, |
108 |
+ |
Float_t iPt1,Float_t iPhi1,Float_t iEta1, |
109 |
+ |
Float_t iPt2,Float_t iPhi2,Float_t iEta2, |
110 |
+ |
const PFMet *iMet , |
111 |
+ |
const PFCandidateCol *iCands,const Vertex *iVertex, |
112 |
+ |
const PFJetCol *iJets , |
113 |
+ |
int iNPV, |
114 |
+ |
Bool_t printDebug); |
115 |
+ |
|
116 |
|
RecoilTools *fRecoilTools; |
117 |
< |
protected: |
117 |
> |
|
118 |
> |
protected: |
119 |
|
TString fPhiMethodName; |
120 |
|
TString fU1MethodName; |
121 |
|
Bool_t fIsInitialized; |
145 |
|
Float_t fNAllJet; |
146 |
|
Float_t fNPV ; |
147 |
|
Float_t fUPhiMVA; |
148 |
< |
|
149 |
< |
TMVA::Reader* fPhiReader; |
150 |
< |
TMVA::Reader* fU1Reader; |
148 |
> |
|
149 |
> |
Float_t* fPhiVals; |
150 |
> |
Float_t* fU1Vals; |
151 |
> |
|
152 |
> |
|
153 |
> |
GBRForest *fPhiReader; |
154 |
> |
GBRForest *fU1Reader; |
155 |
> |
//TMVA::Reader* fPhiReader; |
156 |
> |
//TMVA::Reader* fU1Reader; |
157 |
|
ClassDef(MVAMet,0) |
158 |
|
}; |
159 |
|
} |