c.add_variable("zeppenfeld", 1)
c.add_variable("jets.pt", 2)
c.add_variable("jetPairs.mass", 1)
c.add_variable("jetPairs.dEta", 1)
c.add_variable("jetPairs.dPhi", 1)
###############################################
c.add_data_spectator('muons.pt', 2)
c.add_data_spectator('muPairs.mass', 1)
c.add_data_spectator('muPairs.phi', 1)
c.add_data_spectator('muons.isMediumID', 2)
c.add_data_spectator('jets.phi', 2)
c.add_data_spectator('nJets', 1)
c.add_spectator('muons.pt', 2)
c.add_spectator('muPairs.mass', 1)
c.add_spectator('muPairs.phi', 1)
c.add_spectator('muons.isMediumID', 2)
c.add_spectator('jets.phi', 2)
c.add_spectator('nJets', 1)
c.add_spectator('PU_wgt', 1)
c.add_spectator('GEN_wgt', 1)
c.add_spectator('IsoMu_SF_3', 1)
c.add_spectator('MuID_SF_3', 1)
c.add_spectator('MuIso_SF_3', 1)
c.weigh_by_event(True)
c.add_package("Keras_multi")
c.add_variable('m2eta', 1)
c.add_variable("drmj", 1)
c.add_variable("njets", 1)
c.add_variable("nbjets", 1)
c.add_variable("zepen", 1)
c.add_variable("j1pt", 1)
c.add_variable("j2pt", 1)
c.add_variable("mjj", 1)
c.add_variable("detajj", 1)
c.add_variable("dphijj", 1)
###############################################
c.add_spectator('hmass', 1)
c.add_spectator('weight', 1)
c.weigh_by_event(True)
c.add_package("Keras_multi")
c.add_method("model_50_D2_25_D2_25_D2") # Dropout 0.2
c.train_methods()
print "Training is done: "
print comment
