Make HEPAugFoldYielder work with pT eta phi coordinates #44
Labels
enhancement
New feature or request
good first issue
Good for newcomers
low priority
Not urgent and won't degrade with time
wait
Something that relies on something else occurring first
Current state
HEPAugFoldYielder
applies train-time and test-time augmentation to 3-momenta vectors, but only if they are in Cartesian coordinates (p_x, p_y, p_z). This should ideally be extended to (p_T, eta, phi).Solution
Extend
HEPAugFoldYielder
to run data augmentation over vectors in (p_T, eta, phi) and infer coordinate system from data, rather than requiring user-supplied flag.The text was updated successfully, but these errors were encountered: