Skip to content
GitLab
Projects Groups Snippets
  • /
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
  • Sign in
  • athena athena
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
    • Locked Files
  • Jira
    • Jira
  • Merge requests 146
    • Merge requests 146
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Releases
  • Packages and registries
    • Packages and registries
    • Container Registry
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Code review
    • Repository
  • Activity
  • Graph
  • Jobs
  • Commits
Collapse sidebar
  • atlasatlas
  • athenaathena
  • Merge requests
  • !47137

Adding support for the GNN tagger

  • Review changes

  • Download
  • Email patches
  • Plain diff
Merged Nilotpal Kakati requested to merge nkakati/athena:gnntagger into master Oct 12, 2021
  • Overview 78
  • Commits 27
  • Pipelines 14
  • Changes 19

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 Oct 26, 2021 by Nilotpal Kakati
Assignee
Assign to
Reviewers
Request review from
Time tracking
Source branch: gnntagger