Skip to content

Training on regression labels with NaNs

Nikita Ivvan Pond requested to merge npond/salt:nan_regression into main

Allows for training on regression targets where some labels are None, such as HadronConeExclTruthLabelLxy, which is only defined for jets containing hadrons (mostly B/C jets). This is done simply by only selecting label indices that are non Nan, and calculating the loss on these.

Merge request reports