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_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") # 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
