Exemple #1
0
                       c.dy_label, False)
c.add_file_to_category(ucsd_dy_2018.name, ucsd_dy_2018.path, ucsd_dy_2018.xSec,
                       c.dy_label, False)

c.add_category(c.vbf_label, False)
c.add_file_to_category(ucsd_top_2016.name, ucsd_top_2016.path,
                       ucsd_top_2016.xSec, c.top_label, False)
c.add_file_to_category(ucsd_top_2017.name, ucsd_top_2017.path,
                       ucsd_top_2017.xSec, c.top_label, False)
c.add_file_to_category(ucsd_top_2018.name, ucsd_top_2018.path,
                       ucsd_top_2018.xSec, c.top_label, False)

##########################################################

###  ------   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("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.massWindow = [120,130]

fwk_path = os.environ['FRAMEWORK_PATH']
c.trained_model_path = fwk_path + '/mva/data/DNN27vars_sig_vbf_bkg_dyvbf_dy105To160_ewk105To160_split_60_40_mod10_190820.h5'
c.standartization_path = fwk_path + '/mva/data/DNN27vars_sig_vbf_bkg_dyvbf_dy105To160_ewk105To160_split_60_40_mod10_190820.npy'

##################### Input samples #######################

c.add_category(c.ggh_label, True)
c.add_file_to_category(latinos_ggh_2016.name, latinos_ggh_2016.path,
                       latinos_ggh_2016.xSec, c.ggh_label, False)

##########################################################

###  ------   Caltech variables   ------ ###
c.add_variable("njet")  # replace with softjet5 later
c.add_variable("yll")  # replace with drll
c.add_variable("yll")  # replace with detall
c.add_variable("mjj")
c.add_variable("mjj")  # replace with ptjj
c.add_variable("detajj")  #replace with etajj
c.add_variable("phiCS")  # replace with phijj
c.add_variable('mjj')  # replace with mlljj
c.add_variable('detajj')  #replace with etalljj
c.add_variable("phiCS")  #replace with philljj
c.add_variable("detajj")
c.add_variable("zeppjj")
c.add_variable("drlj")  #min dr(mu,j)
c.add_variable("drlj")  #replace with maxdrlj
c.add_variable("drlj")  #replace with mindrllj
c.add_variable("drlj")  #replace with maxdrllj
# c.add_file_to_category(latinos_ttto2l2nu_2016.name, latinos_ttto2l2nu_2016.path, latinos_ttto2l2nu_2016.xSec, c.top_label, False)

# for TMVA
c.add_signal_file(latinos_ggh_2016.name, latinos_ggh_2016.path + "*part0*",
                  latinos_ggh_2016.xSec, False)
c.add_signal_file(latinos_vbf_2016.name, latinos_vbf_2016.path + "*part0*",
                  latinos_vbf_2016.xSec, False)
c.add_background_file(latinos_dy_2016.name, latinos_dy_2016.path + "*part0*",
                      latinos_dy_2016.xSec, False)
c.add_background_file(latinos_ttto2l2nu_2016.name,
                      latinos_ttto2l2nu_2016.path + "*part0*",
                      latinos_ttto2l2nu_2016.xSec, False)
##########################################################

###  ------   Raffaele's variables   ------ ###
c.add_variable("ptll")
c.add_variable("yll")
c.add_variable("cosThetaCS")
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_file_to_category(ucsd_dy_2016.name, ucsd_dy_2016.path, ucsd_dy_2016.xSec, c.dy_label, False)
c.add_file_to_category(ucsd_dy_2017.name, ucsd_dy_2017.path, ucsd_dy_2017.xSec, c.dy_label, False)
c.add_file_to_category(ucsd_dy_2018.name, ucsd_dy_2018.path, ucsd_dy_2018.xSec, c.dy_label, False)

c.add_category(c.top_label, False)
c.add_file_to_category(ucsd_top_2016.name, ucsd_top_2016.path, ucsd_top_2016.xSec, c.top_label, False)
c.add_file_to_category(ucsd_top_2017.name, ucsd_top_2017.path, ucsd_top_2017.xSec, c.top_label, False)
c.add_file_to_category(ucsd_top_2018.name, ucsd_top_2018.path, ucsd_top_2018.xSec, c.top_label, False)


##########################################################



###  ------   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')