Skip to content

GitLab

Merged
Created by Nilotpal Kakati@nkakatiDeveloper

Adding support for the GNN tagger

This MR adds support to run the GNN based tagger in Athena. The main changes can be summarised as -

  • Adding OnnxUtil which can run onnx models using ONNXRuntime
  • Adding GNNTool which uses OnnxUtil to decorate the btagging objects or the jets
  • Adding DataPrepUtility where we have functions that DL2, DL2HighLevel and GNNTool share. The goal is to store the shared functions in one place and to improve the readability of the Tools themselves.
Edited by Nilotpal Kakati