ONNXRuntime/python/pfHiggsInteractionNet_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from RecoBTag.FeatureTools.pfDeepBoostedJetTagInfos_cfi import pfDeepBoostedJetTagInfos as _pfDeepBoostedJetTagInfos
0004 from RecoBTag.ONNXRuntime.boostedJetONNXJetTagsProducer_cfi import boostedJetONNXJetTagsProducer
0005 from RecoBTag.ONNXRuntime.Parameters.HiggsInteractionNet.V00.pfHiggsInteractionNetPreprocessParams_cfi import pfHiggsInteractionNetPreprocessParams
0006 from RecoBTag.ONNXRuntime.particleNetSonicJetTagsProducer_cfi import particleNetSonicJetTagsProducer as _particleNetSonicJetTagsProducer
0007 from Configuration.ProcessModifiers.particleNetSonicTriton_cff import particleNetSonicTriton
0008
0009 # modify default parameters for tag infos
0010 pfHiggsInteractionNetTagInfos = _pfDeepBoostedJetTagInfos.clone(
0011     min_pt_for_track_properties = 0.95,
0012     min_puppi_wgt = -1,
0013     use_puppiP4 = False,
0014     include_neutrals = False,
0015     sort_by_sip2dsig = True,
0016 )
0017
0018 # nominal Higgs IN
0019 pfHiggsInteractionNetTags = boostedJetONNXJetTagsProducer.clone(
0020     src = 'pfHiggsInteractionNetTagInfos',
0021     preprocessParams = pfHiggsInteractionNetPreprocessParams,
0022     model_path = 'RecoBTag/Combined/data/HiggsInteractionNet/V00/modelfile/model.onnx',
0023     flav_names = [ 'probQCD', 'probHbb' ]
0024 )
0025
0026 particleNetSonicTriton.toReplaceWith(pfHiggsInteractionNetTags, _particleNetSonicJetTagsProducer.clone(
0027     src = 'pfHiggsInteractionNetTagInfos',
0028     preprocess_json = 'RecoBTag/Combined/data/models/higgsInteractionNet/preprocess.json',
0029     Client = cms.PSet(
0030         timeout = cms.untracked.uint32(300),
0031         mode = cms.string("Async"),
0032         modelName = cms.string("higgsInteractionNet"),
0033         modelConfigPath = cms.FileInPath("RecoBTag/Combined/data/models/higgsInteractionNet/config.pbtxt"),
0034         modelVersion = cms.string(""),
0035         verbose = cms.untracked.bool(False),
0036         allowedTries = cms.untracked.uint32(0),
0037         useSharedMemory = cms.untracked.bool(True),
0038         compression = cms.untracked.string(""),
0039     ),
0040     flav_names = pfHiggsInteractionNetTags.flav_names,
0041 ))
0042
0043 # declare all the discriminators
0044 # nominal: probs
0045 _pfHiggsInteractionNetTagsProbs = ['pfHiggsInteractionNetTags:' + flav_name
0046                        for flav_name in pfHiggsInteractionNetTags.flav_names]
0047