Draft: Adding detailed training metric (!262) · Merge requests · atlas-flavor-tagging-tools / algorithms / Salt · GitLab

Wei Sheng Lai requested to merge wlai/salt:add_metric into main Jul 05, 2024

Description

I've added loss calculation of jets according to a pT cut during training.

Related to #7

Adding loss per jet label class
Make this more generic
Update docstring

Review checklist:

CI Passing
Comments addressed
Source branch is up to date with target