ONNXRuntime/python/pfDeepBoostedJet_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from RecoBTag.FeatureTools.pfDeepBoostedJetTagInfos_cfi import pfDeepBoostedJetTagInfos
0004 from RecoBTag.ONNXRuntime.boostedJetONNXJetTagsProducer_cfi import boostedJetONNXJetTagsProducer
0005 from RecoBTag.ONNXRuntime.Parameters.DeepBoostedJet.V02.pfDeepBoostedJetPreprocessParams_cfi import pfDeepBoostedJetPreprocessParams
0006 from RecoBTag.ONNXRuntime.Parameters.DeepBoostedJet.V02.pfMassDecorrelatedDeepBoostedJetPreprocessParams_cfi import pfMassDecorrelatedDeepBoostedJetPreprocessParams
0007 from RecoBTag.ONNXRuntime.pfDeepBoostedDiscriminatorsJetTags_cfi import pfDeepBoostedDiscriminatorsJetTags
0008 from RecoBTag.ONNXRuntime.pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags_cfi import pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags
0009
0010 # nominal DeepAK8
0011 pfDeepBoostedJetTags = boostedJetONNXJetTagsProducer.clone(
0012     preprocessParams = pfDeepBoostedJetPreprocessParams,
0013     model_path = 'RecoBTag/Combined/data/DeepBoostedJet/V02/full/resnet.onnx',
0014 )
0015
0016 # mass-decorrelated DeepAK8
0017 pfMassDecorrelatedDeepBoostedJetTags = boostedJetONNXJetTagsProducer.clone(
0018     preprocessParams = pfMassDecorrelatedDeepBoostedJetPreprocessParams,
0019     model_path = 'RecoBTag/Combined/data/DeepBoostedJet/V02/decorrelated/resnet.onnx',
0020 )
0021
0022 from CommonTools.PileupAlgos.Puppi_cff import puppi
0023 from CommonTools.RecoAlgos.primaryVertexAssociation_cfi import primaryVertexAssociation
0024
0025 # This task is not used, useful only if we run DeepFlavour from RECO
0026 # jets (RECO/AOD)
0027 pfDeepBoostedJetTask = cms.Task(puppi, primaryVertexAssociation,
0028                              pfDeepBoostedJetTagInfos, pfDeepBoostedJetTags, pfMassDecorrelatedDeepBoostedJetTags,
0029                              pfDeepBoostedDiscriminatorsJetTags, pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags)
0030
0031 # declare all the discriminators
0032 # nominal: probs
0033 _pfDeepBoostedJetTagsProbs = ['pfDeepBoostedJetTags:' + flav_name
0034                               for flav_name in pfDeepBoostedJetTags.flav_names]
0035 # nominal: meta-taggers
0036 _pfDeepBoostedJetTagsMetaDiscrs = ['pfDeepBoostedDiscriminatorsJetTags:' + disc.name.value()
0037                                    for disc in pfDeepBoostedDiscriminatorsJetTags.discriminators]
0038
0039 # mass-decorrelated: probs
0040 _pfMassDecorrelatedDeepBoostedJetTagsProbs = ['pfMassDecorrelatedDeepBoostedJetTags:' + flav_name
0041                                               for flav_name in pfMassDecorrelatedDeepBoostedJetTags.flav_names]
0042 # mass-decorrelated: meta-taggers
0043 _pfMassDecorrelatedDeepBoostedJetTagsMetaDiscrs = ['pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags:' + disc.name.value()
0044                                    for disc in pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags.discriminators]
0045
0046 _pfDeepBoostedJetTagsAll = _pfDeepBoostedJetTagsProbs + _pfDeepBoostedJetTagsMetaDiscrs + \
0047     _pfMassDecorrelatedDeepBoostedJetTagsProbs + _pfMassDecorrelatedDeepBoostedJetTagsMetaDiscrs