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

make training features configurable #20

Open jpata opened 1 month ago

jpata commented 1 month ago

For all 3 models, both input_dim of each model and the unpack_... functions (https://github.com/HEP-KBFI/ml-tau-en-reg/blob/main/enreg/scripts/trainModel.py#L46-L71) need to take into account three possible feature sets:

This should be easily configurable with a single flag (e.g. feature_set=[kinematics], feature_set=[kinematics,pdgs], feature_set=[kinematics,pdgs,lifetimes], to enable scanning across the options in https://github.com/HEP-KBFI/ml-tau-en-reg/issues/17.

jpata commented 2 weeks ago

possible next task for @HardiVv.