def main():
infiles = sys.argv[1:]
print infiles
mvas = dict()
# mvas[13] = setup_mva(mvaname, STPOL_DIR + "/mvatools/weights/stop_mu_BDT.weights.xml")
# mvas[11] = setup_mva(mvaname, STPOL_DIR + "/mvatools/weights/stop_ele_BDT.weights.xml")
mvas[13] = setup_mva(mvaname, STPOL_DIR + "/mvatools/weights/stop_mu_BDT_with_top_mass_eta_lj_C_mu_pt_mt_mu_met_mass_bj_pt_bj_mass_lj.weights.xml")
mvas[11] = setup_mva(mvaname, STPOL_DIR + "/mvatools/weights/stop_ele_BDT_with_top_mass_C_eta_lj_el_pt_mt_el_pt_bj_mass_bj_met_mass_lj.weights.xml")
varmaps = dict()
varmap_general = {
"mass_bj":"bjet_mass",
"eta_lj":"ljet_eta",
"mass_lj":"ljet_mass",
"pt_bj":"bjet_pt",
"C":"C_with_nu",
}
varmaps[13] = {"mt_mu":"mtw", "mu_pt":"lepton_pt"}
varmaps[11] = {"mt_el":"mtw", "el_pt":"lepton_pt"}
for (k, v) in varmaps.items():
varmaps[k].update(varmap_general)
mva_loop_lepton_separate(mvaname, infiles, mvas, varmaps)
def main():
infiles = sys.argv[1:]
print infiles
mvas = dict()
mvas[13] = setup_mva(mvaname, STPOL_DIR + "/src/qcd_mva/weights/anti_QCD_MVA_07_04_final2_mu.weights.xml")
mvas[11] = setup_mva(mvaname, STPOL_DIR + "/src/qcd_mva/weights/anti_QCD_MVA_07_04_final2_ele.weights.xml")
varmaps = dict()
varmaps[13] = {"mu_mtw":"mtw", "c":"C"}
varmaps[11] = {"ele_mtw":"mtw", "c":"C"}
mva_loop_lepton_separate(mvaname, infiles, mvas, varmaps)
def main():
mvaname = sys.argv[1]
weightname = sys.argv[2]
infiles = sys.argv[3:]
print infiles
mvas = dict()
mvas[13] = setup_mva(mvaname, STPOL_DIR + "/src/qcd_mva/weights/%s_mu.weights.xml" % weightname)
mvas[11] = setup_mva(mvaname, STPOL_DIR + "/src/qcd_mva/weights/%s_ele.weights.xml" % weightname)
varmaps = dict()
varmaps[13] = {"mu_mtw":"mtw", "c":"C"}
varmaps[11] = {"ele_mtw":"mtw", "c":"C"}
mva_loop_lepton_separate(mvaname, infiles, mvas, varmaps, weightname)
