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
@@ -61,7 +61,7 @@ def createTauConfigFlags():
tau_cfg.addFlag("Tau.TauJetDeepSetConfig_v2", ["tauid_1p_R22_dpst_noTrackScore.json", "tauid_2p_R22_dpst_noTrackScore.json", "tauid_3p_R22_dpst_noTrackScore.json"])
tau_cfg.addFlag("Tau.TauJetDeepSetWP_v2", ["model_1p_R22_dpst_noTrackScore.root", "model_2p_R22_dpst_noTrackScore.root", "model_3p_R22_dpst_noTrackScore.root"])
# GNTau ID tune file (need to add another version for noAux)
tau_cfg.addFlag("Tau.TauGNNConfig", ["GNTau_preval.onnx"])
tau_cfg.addFlag("Tau.TauGNNConfig", ["GNTau_preval.onnx","GNTau_noAux_preval.onnx"])
# PanTau config flags
from PanTauAlgs.PanTauConfigFlags import createPanTauConfigFlags
Loading