Unify training code, improve visualization

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

Closed jpata closed 4 months ago

jpata commented 4 months ago

trainModel.py replaces trainParticleTransformer.py, trainLorentzNet.py, trainSimpleDNN.py
all 3 models (SimpleDNN, LorentzNet, ParticleTransformer) can be trained for all 3 tasks (binary classification, jet regression, decay mode classification)
fixed bugs and improved visualization in tensorboard

decay mode:

jet regression:

binary classification: