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("TMVA")
c.add_method("BDTG_UCSD")
# c.add_method("BDTG_MIT")
c.weigh_by_event(True)
c.train_methods()
print "Training is done: "
print comment
print "Output saved to:"
print c.outPath
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_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
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_method("model_50_D2_25_D2_25_D2")
c.add_method("model_50_D2_25_D2_25_D2_25_D2") # deeper
c.add_method("model_100_D2_50_D2_50_D2") # wider
c.add_method("model_100_D2_50_D2_50_D2_50_D2") # deeper & wider
c.train_methods()
print "Training is done: "
print comment
print "Output saved to:"
print c.outPath
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("TMVA")
c.add_method("BDTG_UF_v1")
# c.add_method("BDTG_MIT")
c.weigh_by_event(True)
c.train_methods()
print "Training is done: "
print comment
print "Output saved to:"
print c.outPath
c.add_background(b[0], bkg_path+b[1], b[2])
c.set_tree_path(treePath)
c.add_variable("muPairs.pt", 1) #second argument is the number of objects considered
c.add_variable("muPairs.eta", 1)
c.add_variable("muPairs.phi", 1)
c.add_variable("muPairs.dEta", 1)
c.add_variable("muPairs.dPhi", 1)
c.add_variable("muons.eta", 2)
c.add_variable("met.pt", 1)
c.add_variable("nJets", 1)
c.add_variable("nJetsCent", 1)
# c.add_variable("nJetsFwd", 1)
c.add_variable("nBMed", 1)
# c.add_variable("jets.eta", 2)
c.add_variable("jetPairs.dEta", 2)
c.add_variable("jetPairs.mass", 2)
# c.add_variable("muons.pt", 2) #two muons
# c.add_variable("muons.phi", 2)
c.set_year("2017")
c.add_package("TMVA")
c.add_transf("N,G,P")
c.add_method("BDTG_MIT")
# c.add_method("BDTG_UF_v1")
# c.add_package("Keras")
c.train_methods()
c.add_spectator('MuIso_SF_4', 1)
c.weigh_by_event(True)
c.set_year("2016")
# c.add_package("TMVA")
# c.add_transf("N,G,P")
# c.add_method("BDTG_UF_v1")
c.add_package("Keras")
# c.add_method("UCSD_model") # 50_D2
# c.add_method("model_50_25") # no Dropout
# c.add_method("model_50_D1_25_D1") # Dropout 0.1
# c.add_method("model_50_D2_25_D2") # Dropout 0.2
c.custom_loss = True
c.add_method("model_50_D2_25_D2_kldiv0")
c.add_method("model_50_D2_25_D2_kldiv1")
c.add_method("model_50_D2_25_D2_kldiv2")
# c.add_method("model_50_D2_25_D2_kldiv3")
# c.add_method("model_50_D2_25_D2_kldiv4")
# c.add_method("model_50_D2_25_D2_kldiv5")
# c.add_method("model_50_D3_25_D3") # Dropout 0.3
# c.add_method("model_50_D1_25_D1_10_D1") # Dropout 0.1
# c.add_method("model_50_25_25") # no Dropout
# c.add_method("model_50_D1_25_D1_25_D1") # Dropout 0.1
# c.add_method("model_50_D1_50_D1_50_D1") # Dropout 0.1
# c.add_method("model_50_D2_25_D2_10_D2") # Dropout 0.2
# c.add_method("model_50_D2_25_D2_25_D2") # Dropout 0.2
# c.add_method("model_50_D3_25_D3_25_D3") # Dropout 0.3
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.add_spectator('IsoMu_SF_4', 1)
c.add_spectator('MuID_SF_4', 1)
c.add_spectator('MuIso_SF_4', 1)
c.weigh_by_event(True)
c.set_year("2016")
# c.add_package("TMVA")
# c.add_transf("N,G,P")
# c.add_method("BDTG_UF_v1")
c.add_package("Keras")
c.add_method("UCSD_model")
c.add_method("model_50_D_25_D")
c.train_methods()
## HIG-17-019 variables:
# c.add_variable("muPairs.pt", 1)
# c.add_variable("muPairs.eta", 1)
# c.add_variable("muPairs.dEta", 1)
# c.add_variable("muPairs.dPhi", 1)
# c.add_variable("met.pt", 1)
# c.add_variable("nJetsCent", 1)
# c.add_variable("nJetsFwd", 1)
# c.add_variable("nBMed", 1)
# c.add_variable("jets.eta", 2)
# c.add_variable("jetPairs.dEta", 2)
