Skip to content
GitLab

Adding support for the GNN tagger

Merged Nilotpal Kakati requested to merge nkakati/athena:gnntagger into master

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