def __init__(self, signame, bkgname, variables, cuts, label, treeName, weightloc='weights', MVAMethod="BDTG"): self._f_sig = ROOT.TFile(signame) self._f_bkg = ROOT.TFile(bkgname) self._t_sig = self._f_sig.Get(treeName) self._t_bkg = self._f_bkg.Get(treeName) self._discVariables = variables #,"MHT","nJets_30"]; #tmva cut self._cutstring = "(" cutctr = 0 for cut in cuts: self._cutstring += "(" + cut[0] + " > " + str( cut[1]) + ") && (" + cut[0] + " < " + str(cut[2]) + ")" if cutctr < len(cuts) - 1: self._cutstring += "&&" cutctr += 1 self._cutstring += ")" print "cutstring = ", self._cutstring # self._spectatorVariables = ["lheWeight"]; self._spectatorVariables = [] self._trees = [self._t_sig, self._t_bkg] self._bdt = TMVAhelper("MVA_" + label, self._discVariables, self._trees, self._spectatorVariables, weightloc) print ";;weightloc = ", weightloc