Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2024-11-05 05:20:43

0001 import FWCore.ParameterSet.Config as cms
0002 
0003 from RecoBTag.FeatureTools.pfGlobalParticleTransformerAK8TagInfos_cfi import pfGlobalParticleTransformerAK8TagInfos as _pfGlobalParticleTransformerAK8TagInfos
0004 from RecoBTag.ONNXRuntime.boostedJetONNXJetTagsProducer_cfi import boostedJetONNXJetTagsProducer
0005 
0006 pfGlobalParticleTransformerAK8TagInfos = _pfGlobalParticleTransformerAK8TagInfos.clone(
0007     use_puppiP4 = False
0008 )
0009 
0010 pfGlobalParticleTransformerAK8JetTags = boostedJetONNXJetTagsProducer.clone(
0011     src = 'pfGlobalParticleTransformerAK8TagInfos',
0012     preprocess_json = 'RecoBTag/Combined/data/GlobalParticleTransformerAK8/PUPPI/V03/preprocess.json',
0013     model_path = 'RecoBTag/Combined/data/GlobalParticleTransformerAK8/PUPPI/V03/model.onnx',
0014     flav_names = [
0015         'probXbb', 'probXcc', 'probXcs', 'probXqq', 'probXtauhtaue', 'probXtauhtaum', 'probXtauhtauh', 'probXWW4q', 'probXWW3q', 'probXWWqqev', 'probXWWqqmv', 'probTopbWqq', 'probTopbWq', 'probTopbWev', 'probTopbWmv', 'probTopbWtauhv', 'probQCD', 'massCorrX2p', 'massCorrGeneric', 'probWithMassTopvsQCD', 'probWithMassWvsQCD', 'probWithMassZvsQCD'
0016     ] + ['hidNeuron' + str(i).zfill(3) for i in range(256)],
0017     debugMode = False,
0018 )
0019 
0020 from CommonTools.PileupAlgos.Puppi_cff import puppi
0021 from CommonTools.RecoAlgos.primaryVertexAssociation_cfi import primaryVertexAssociation
0022 
0023 # This task is not used, useful only if we run it from RECO jets (RECO/AOD)
0024 pfGlobalParticleTransformerAK8Task = cms.Task(puppi, primaryVertexAssociation, pfGlobalParticleTransformerAK8TagInfos, pfGlobalParticleTransformerAK8JetTags)
0025 
0026 # declare all the discriminators
0027 
0028 # probs
0029 _pfGlobalParticleTransformerAK8JetTagsProbs = ['pfGlobalParticleTransformerAK8JetTags:' + flav_name for flav_name in pfGlobalParticleTransformerAK8JetTags.flav_names]
0030 
0031 # meta-taggers
0032 _pfGlobalParticleTransformerAK8JetTagsMetaDiscrs = []
0033 
0034 _pfGlobalParticleTransformerAK8JetTagsAll = _pfGlobalParticleTransformerAK8JetTagsProbs + _pfGlobalParticleTransformerAK8JetTagsMetaDiscrs