Skip to content
Snippets Groups Projects

Implementation of ONNX model score computation and decoration for Taus (GNTau)

Merged Nadav Michael Tamir requested to merge ntamir/athena:master into main
9 files
+ 61
58
Compare changes
  • Side-by-side
  • Inline
Files
9
@@ -103,7 +103,7 @@ def AddTauIDDecorationCfg(flags, **kwargs):
kwargs.setdefault("evetoFix", True)
kwargs.setdefault("DeepSetID", True)
kwargs.setdefault("GNTauID", True)
kwargs.setdefault("GNNTauID", True)
kwargs.setdefault("TauContainerName", "TauJets")
kwargs.setdefault("prefix", kwargs['TauContainerName'])
@@ -123,8 +123,11 @@ def AddTauIDDecorationCfg(flags, **kwargs):
# R22 DeepSet tau ID tune without track RNN scores
tools.append( acc.popToolsAndMerge(tauTools.TauJetDeepSetEvaluatorCfg(flags, version="v2")) )
tools.append( acc.popToolsAndMerge(tauTools.TauWPDecoratorJetDeepSetCfg(flags, version="v2")) )
if kwargs['GNNTauID']:
# Add in GNTau!
tools.append( acc.popToolsAndMerge(tauTools.TauGNNEvaluatorCfg(flags)) )
tools.append( acc.popToolsAndMerge(tauTools.TauWPDecoratorGNNCfg(flags)) )
if tools:
for tool in tools:
Loading