feat: add DeepJet inputs

cms-jet / PFNano

NanoAOD framework for testing jet algorithms.

7 stars 54 forks source link

feat: add DeepJet inputs #39

Closed AnnikaStein closed 2 years ago

AnnikaStein commented 2 years ago

Add custom producer to get information for 25 cpfs, 25 npfs and 4 SVs (similar to JetConstituentTableProducer), but store these in the already existing Jet table (flat, no new collections) - if requested via add_DeepJet and keep_inputs
Include full output nodes for DeepJet (if requested via add_DeepJet) - I guess #35 is related
Second new option add_DeepJet_noclip, which adds more variables on top of add_DeepJet: uses raw quantities without any preprocessing by catch_infs_and_bound (variables are not clipped / scaled)

@andrzejnovak

AnnikaStein commented 2 years ago

Access DeepJet inputs via TagInfoCollection in a new, custom producer, similar to JetConstituentTableProducer.cc, but using the logic of DeepFlavourONNXJetTagsProducer.cc
No more recalculation of features necessary (but could of course revert to the old way and compare - my personal checks showed they are in good agreement) - see this notebook for details (and scroll all the way down, I'm sorry)
Minor clean up in other files

Note: size and descriptions for 5000 events when DeepJet inputs (and output probabilities) are added. (Cf. old size and descriptions from recalculating variables. See also the same number of events, without DeepJet here: size and description.)

AnnikaStein commented 2 years ago

All requested changes should be in now, but as I'm only Author, not Collaborator, I think someone else has to merge this PR.