c.add_variable("phiCS")
c.add_variable("MET_pt")
c.add_variable("pt1omll")
c.add_variable("pt2omll")
c.add_variable('Muon_eta', 2)
# c.add_variable("drmj")
c.add_variable("nJet")
# c.add_variable("nbjets")
c.add_variable("zeppjj")
c.add_variable("Jet_pt", 2)
c.add_variable("mjj")
c.add_variable("detajj")
# c.add_variable("dphijj")
###############################################
c.add_spectator('mll')
c.add_spectator('XSWeight')
c.add_spectator('SFweight')
c.add_spectator('GenLepMatch')
c.add_spectator('METFilter_MC')
c.weigh_by_event(True)
c.add_package("TMVA")
c.add_method("BDTG_UCSD") # Dropout 0.2
# c.add_package("Keras")
# c.add_method("model_50_D2_25_D2_25_D2") # Dropout 0.2
c.train_methods()
c.add_variable("met")
c.add_variable("m1ptOverMass")
c.add_variable("m2ptOverMass")
c.add_variable('m1eta')
c.add_variable('m2eta')
c.add_variable("drmj")
c.add_variable("njets")
c.add_variable("nbjets")
c.add_variable("zepen")
c.add_variable("j1pt")
c.add_variable("j2pt")
c.add_variable("mjj")
c.add_variable("detajj")
c.add_variable("dphijj")
###############################################
c.add_spectator('hmass')
c.add_spectator('weight')
c.weigh_by_event(True)
c.add_package("Keras")
c.add_method("model_50_D2_25_D2_25_D2") # Dropout 0.2
c.train_methods()
print "Training is done: "
print comment
print "Output saved to:"
print c.outPath
### ------ Raffaele's variables ------ ###
c.add_variable("hmmpt")
c.add_variable("hmmrap")
c.add_variable("hmmthetacs")
c.add_variable("hmmphics")
c.add_variable("met")
c.add_variable("m1ptOverMass")
c.add_variable("m2ptOverMass")
c.add_variable('m1eta')
c.add_variable('m2eta')
c.add_variable("j1pt")
c.add_variable("j1eta")
c.add_variable("nbjets")
###############################################
c.add_spectator('hmass')
c.add_spectator('weight')
c.add_spectator("njets")
c.weigh_by_event(True)
c.add_package("Keras")
c.add_method("model_50_D2_25_D2_25_D2") # Dropout 0.2
c.train_methods()
print "Training is done: "
print comment
print "Output saved to:"
print c.outPath