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
0024 pfGlobalParticleTransformerAK8Task = cms.Task(puppi, primaryVertexAssociation, pfGlobalParticleTransformerAK8TagInfos, pfGlobalParticleTransformerAK8JetTags)
0025
0026
0027
0028
0029 _pfGlobalParticleTransformerAK8JetTagsProbs = ['pfGlobalParticleTransformerAK8JetTags:' + flav_name for flav_name in pfGlobalParticleTransformerAK8JetTags.flav_names]
0030
0031
0032 _pfGlobalParticleTransformerAK8JetTagsMetaDiscrs = []
0033
0034 _pfGlobalParticleTransformerAK8JetTagsAll = _pfGlobalParticleTransformerAK8JetTagsProbs + _pfGlobalParticleTransformerAK8JetTagsMetaDiscrs