HEP-KBFI / ml-tau-en-reg

Reconstructing tau leptons using various machine learning techniques
GNU General Public License v3.0
0 stars 1 forks source link

feature sets for other models, improve memory usage, cleanup old stuff #65

Closed jpata closed 1 month ago

jpata commented 1 month ago
jpata commented 1 month ago

Training works and gives losses that are expected with the PR

Screenshot 2024-07-11 at 09 50 26 Screenshot 2024-07-11 at 09 50 39 Screenshot 2024-07-11 at 09 50 01
jpata commented 1 month ago

before and after this PR, LorentzNet has equivalent performance image

while ParticleTransformer is now better (probably due to removing the SequenceTrimmer): image

for SimpleDNN, we now revert back to the cand_ParT_features (in #62, the default was changed to kinematics+ParT+lifetimes), which is better: image