ATLAS Offline Software
Loading...
Searching...
No Matches
Functions
BoostedJetTaggerConfig Namespace Reference

Functions

 qgTagAlgCfg (configFlags, **kwargs)
 WZTagAlgCfg (configFlags, **kwargs)
 TopTagAlgCfg (configFlags, **kwargs)
 BJTToolCfg (flags, **kwargs)
 BJTSFToolCfg (flags, **kwargs)

Function Documentation

◆ BJTSFToolCfg()

BoostedJetTaggerConfig.BJTSFToolCfg ( flags,
** kwargs )

Definition at line 167 of file BoostedJetTaggerConfig.py.

167def BJTSFToolCfg(flags, **kwargs):
168
169 acc = ComponentAccumulator()
170
171 kwargs.setdefault("OutputLevel", DEBUG)
172
173 acc.setPrivateTools(CompFactory.BJT.ScaleFactors(**kwargs))
174
175 return acc

◆ BJTToolCfg()

BoostedJetTaggerConfig.BJTToolCfg ( flags,
** kwargs )

Definition at line 150 of file BoostedJetTaggerConfig.py.

150def BJTToolCfg(flags, **kwargs):
151
152 acc = ComponentAccumulator()
153
154 kwargs.setdefault("CalibArea", "Local")
155 kwargs.setdefault("IsMC", 1)
156 kwargs.setdefault("OutputLevel", DEBUG)
157
158 if flags['tagger'] == 'WZ':
159 acc.setPrivateTools(CompFactory.SmoothedWZTagger(**kwargs))
160 elif flags['tagger'] == 'Top':
161 acc.setPrivateTools(CompFactory.SmoothedTopTagger(**kwargs))
162 elif flags['tagger'] == 'qg':
163 acc.setPrivateTools(CompFactory.BJT.qgTagger(**kwargs))
164
165 return acc
166

◆ qgTagAlgCfg()

BoostedJetTaggerConfig.qgTagAlgCfg ( configFlags,
** kwargs )

Definition at line 8 of file BoostedJetTaggerConfig.py.

8def qgTagAlgCfg(configFlags, **kwargs):
9
10 acc = ComponentAccumulator()
11
12 # add syst service
13 sysService = CompFactory.CP.SystematicsSvc("SystematicsSvc")
14 acc.addService(sysService)
15
16 # jet container name
17 jets_container = "AntiKt4EMPFlowJets"
18
19 # BJT config file
20 config_file = kwargs['cfg_file']
21
22 # run
23 run = kwargs['run']
24
25 # input WP file
26 wps_file = 'QGTagger_WPs_' + run + '.root'
27 wps_histo = "QGTransformer_ConstScore__" + kwargs['WP'] + "WP"
28
29 # configure the histogram reader
30 tool_args = {}
31
32 histo2D = HistoInputCfg(configFlags, "histo2D",
33 inputFile = kwargs['calib_path'] + '/' + wps_file,
34 histName = wps_histo,
35 varX = "pt", varY = "eta",
36 InterpType="None")
37
38 tool_args.setdefault("HistoReader2D", histo2D)
39
40 # configure the kinematic range
41 tool_args.setdefault("jetPtMin", 20.)
42 tool_args.setdefault("jetPtMax", 2000.)
43 tool_args.setdefault("jetEtaMax", 4.5)
44
45 # pick the BJT tool
46 tool_args.setdefault("ConfigFile", config_file)
47 tool_args.setdefault("ContainerName", jets_container)
48 tool = acc.popToolsAndMerge(BJTToolCfg(kwargs, **tool_args))
49
50 # sf tool cfg
51 if kwargs['addSFs']:
52 # histo name
53 sfs_histo = "QGTransformer_ConstScore__" + kwargs['WP'] + "WP"
54
55 # configure the histogram reader
56 sfs_tool_args = {}
57
58 histo2D = {}
59 for parton in ['quark', 'gluon']:
60 for eff in ['eff', 'ineff']:
61 # input SF file
62 sfs_file = 'SF_' + eff + '_' + parton + '_' + run + '.root'
63
64 histo2D[parton + '_' + eff] = HistoInputCfg(configFlags, "histo2D",
65 inputFile = kwargs['calib_path'] + '/' + sfs_file,
66 histName = sfs_histo,
67 varX = "pt", varY = "eta",
68 InterpType="None")
69 sfs_tool_args.setdefault("HistoReader2D_" + parton + "_" + eff, histo2D[parton + '_' + eff])
70
71 # pick the BJT tool
72 sfs_tool_args.setdefault("JetContainer", jets_container)
73 sfs_tool_args.setdefault("TaggedName", "QGTransformer_Tagged")
74 sfs_tool_args.setdefault("Efficiency", "QGTransformer_Efficiency")
75 sfs_tool_args.setdefault("Inefficiency", "QGTransformer_Inefficiency")
76 sfs_tool_args.setdefault("truthLabelName", "PartonTruthLabelID")
77 sfs_tool_args.setdefault("jetPtMin", 20.)
78 sfs_tool_args.setdefault("jetPtMax", 2000.)
79 sfs_tool_args.setdefault("jetEtaMax", 4.5)
80 sfs_tool = acc.popToolsAndMerge(BJTSFToolCfg(kwargs, **sfs_tool_args))
81
82 # configure the BJT algo
83 algo_args = {}
84 algo_args.setdefault("tagger", tool)
85 if kwargs['addSFs']:
86 algo_args.setdefault("scalefactor", sfs_tool)
87 algo_args.setdefault("jets", jets_container)
88 acc.addEventAlgo(CompFactory.BJT.BoostedJetTaggerAlg("qgAlg", **algo_args))
89
90 return acc
91

◆ TopTagAlgCfg()

BoostedJetTaggerConfig.TopTagAlgCfg ( configFlags,
** kwargs )

Definition at line 121 of file BoostedJetTaggerConfig.py.

121def TopTagAlgCfg(configFlags, **kwargs):
122
123 acc = ComponentAccumulator()
124
125 # add syst service
126 sysService = CompFactory.CP.SystematicsSvc("SystematicsSvc")
127 acc.addService(sysService)
128
129 # jet container name
130 jets_container = "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets"
131
132 # BJT config file
133 config_file = kwargs['cfg_file']
134 config_file += "TopTagger_AntiKt10UFOSoftDrop_" + kwargs['generation'] + "_FixSigEff" + kwargs['WP'] + ".dat"
135
136 # pick the BJT tool
137 tool_args = {}
138 tool_args.setdefault("ConfigFile", config_file)
139 tool_args.setdefault("ContainerName", jets_container)
140 tool = acc.popToolsAndMerge(BJTToolCfg(kwargs, **tool_args))
141
142 # configure the BJT algo
143 algo_args = {}
144 algo_args.setdefault("tagger", tool)
145 algo_args.setdefault("jets", jets_container)
146 acc.addEventAlgo(CompFactory.BJT.BoostedJetTaggerAlg("TopTagAlg", **algo_args))
147
148 return acc
149

◆ WZTagAlgCfg()

BoostedJetTaggerConfig.WZTagAlgCfg ( configFlags,
** kwargs )

Definition at line 92 of file BoostedJetTaggerConfig.py.

92def WZTagAlgCfg(configFlags, **kwargs):
93
94 acc = ComponentAccumulator()
95
96 # add syst service
97 sysService = CompFactory.CP.SystematicsSvc("SystematicsSvc")
98 acc.addService(sysService)
99
100 # jet container name
101 jets_container = "AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets"
102
103 # BJT config file
104 config_file = kwargs['cfg_file']
105 config_file += "WTagger_AntiKt10UFOSoftDrop_" + kwargs['generation'] + "_FixSigEff" + kwargs['WP'] + ".dat"
106
107 # pick the BJT tool
108 tool_args = {}
109 tool_args.setdefault("ConfigFile", config_file)
110 tool_args.setdefault("ContainerName", jets_container)
111 tool = acc.popToolsAndMerge(BJTToolCfg(kwargs, **tool_args))
112
113 # configure the BJT algo
114 algo_args = {}
115 algo_args.setdefault("tagger", tool)
116 algo_args.setdefault("jets", jets_container)
117 acc.addEventAlgo(CompFactory.BJT.BoostedJetTaggerAlg("WTagAlg", **algo_args))
118
119 return acc
120
Generated on for ATLAS Offline Software by doxygen 1.16.1