import FWCore.ParameterSet.Config as cms
from RecoBTag.FeatureTools.pfDeepBoostedJetTagInfos_cfi import pfDeepBoostedJetTagInfos as _pfDeepBoostedJetTagInfos
from RecoBTag.ONNXRuntime.pfDeepBoostedJetTags_cfi import pfDeepBoostedJetTags as _pfDeepBoostedJetTags
from RecoBTag.ONNXRuntime.Parameters.HiggsInteractionNet.V00.pfHiggsInteractionNetPreprocessParams_cfi import pfHiggsInteractionNetPreprocessParams
# modify default parameters for tag infos
pfHiggsInteractionNetTagInfos = _pfDeepBoostedJetTagInfos.clone(
min_pt_for_track_properties=0.95,
min_puppi_wgt=-1,
use_puppiP4=False,
include_neutrals=False,
sort_by_sip2dsig=True,
)
# nominal Higgs IN
pfHiggsInteractionNetTags = _pfDeepBoostedJetTags.clone(
src='pfHiggsInteractionNetTagInfos',
preprocessParams=pfHiggsInteractionNetPreprocessParams,
model_path='RecoBTag/Combined/data/HiggsInteractionNet/V00/IN.onnx',
flav_names=['probQCD', 'probHbb'])
# declare all the discriminators
# nominal: probs
_pfHiggsInteractionNetTagsProbs = [
'pfHiggsInteractionNetTags:' + flav_name
for flav_name in pfHiggsInteractionNetTags.flav_names
]
import FWCore.ParameterSet.Config as cms
from RecoBTag.FeatureTools.pfDeepBoostedJetTagInfos_cfi import pfDeepBoostedJetTagInfos
from RecoBTag.ONNXRuntime.boostedJetONNXJetTagsProducer_cfi import boostedJetONNXJetTagsProducer
from RecoBTag.ONNXRuntime.pfParticleNetDiscriminatorsJetTags_cfi import pfParticleNetDiscriminatorsJetTags
from RecoBTag.ONNXRuntime.pfMassDecorrelatedParticleNetDiscriminatorsJetTags_cfi import pfMassDecorrelatedParticleNetDiscriminatorsJetTags
pfParticleNetTagInfos = pfDeepBoostedJetTagInfos.clone(
use_puppiP4 = False
)
pfParticleNetJetTags = boostedJetONNXJetTagsProducer.clone(
src = 'pfParticleNetTagInfos',
preprocess_json = 'RecoBTag/Combined/data/ParticleNetAK8/General/V01/preprocess.json',
model_path = 'RecoBTag/Combined/data/ParticleNetAK8/General/V01/particle-net.onnx',
flav_names = ["probTbcq", "probTbqq", "probTbc", "probTbq", "probTbel", "probTbmu", "probTbta",
"probWcq", "probWqq", "probZbb", "probZcc", "probZqq", "probHbb", "probHcc",
"probHqqqq", "probQCDbb", "probQCDcc", "probQCDb", "probQCDc", "probQCDothers"],
)
pfMassDecorrelatedParticleNetJetTags = boostedJetONNXJetTagsProducer.clone(
src = 'pfParticleNetTagInfos',
preprocess_json = 'RecoBTag/Combined/data/ParticleNetAK8/MD-2prong/V01/preprocess.json',
model_path = 'RecoBTag/Combined/data/ParticleNetAK8/MD-2prong/V01/particle-net.onnx',
flav_names = ["probXbb", "probXcc", "probXqq", "probQCDbb", "probQCDcc",
"probQCDb", "probQCDc", "probQCDothers"],
)
pfParticleNetMassRegressionJetTags = boostedJetONNXJetTagsProducer.clone(
src = 'pfParticleNetTagInfos',
preprocess_json = 'RecoBTag/Combined/data/ParticleNetAK8/MassRegression/V01/preprocess.json',
import FWCore.ParameterSet.Config as cms
from RecoBTag.FeatureTools.pfDeepBoostedJetTagInfos_cfi import pfDeepBoostedJetTagInfos
from RecoBTag.ONNXRuntime.boostedJetONNXJetTagsProducer_cfi import boostedJetONNXJetTagsProducer
from RecoBTag.ONNXRuntime.particleNetSonicJetTagsProducer_cfi import particleNetSonicJetTagsProducer as _particleNetSonicJetTagsProducer
from RecoBTag.ONNXRuntime.pfParticleNetAK4DiscriminatorsJetTags_cfi import pfParticleNetAK4DiscriminatorsJetTags
from Configuration.ProcessModifiers.particleNetSonicTriton_cff import particleNetSonicTriton
from Configuration.ProcessModifiers.particleNetPTSonicTriton_cff import particleNetPTSonicTriton
pfParticleNetAK4TagInfos = pfDeepBoostedJetTagInfos.clone(
jets = "ak4PFJetsCHS",
jet_radius = 0.4,
min_jet_pt = 15,
min_puppi_wgt = -1,
use_puppiP4 = False,
)
pfParticleNetAK4TagInfosForRECO = pfParticleNetAK4TagInfos.clone(
min_jet_pt = 25,
max_jet_eta = 2.5,
)
pfParticleNetAK4JetTags = boostedJetONNXJetTagsProducer.clone(
src = 'pfParticleNetAK4TagInfos',
preprocess_json = 'RecoBTag/Combined/data/ParticleNetAK4/CHS/V00/preprocess.json',
model_path = 'RecoBTag/Combined/data/ParticleNetAK4/CHS/V00/modelfile/model.onnx',
flav_names = ["probb", "probbb", "probc", "probcc", "probuds", "probg", "probundef", "probpu"],
)
pfParticleNetAK4JetTagsForRECO = pfParticleNetAK4JetTags.clone(
src = 'pfParticleNetAK4TagInfosForRECO',