File indexing completed on 2024-10-01 05:02:15
0001 import FWCore.ParameterSet.Config as cms
0002 from RecoJets.JetProducers.PileupJetIDCutParams_cfi import *
0003
0004
0005 full_81x_chs = cms.PSet(
0006 impactParTkThreshold = cms.double(1.),
0007 cutBased = cms.bool(False),
0008 etaBinnedWeights = cms.bool(True),
0009 tmvaMethod = cms.string("JetIDMVAHighPt"),
0010 version = cms.int32(-1),
0011 nEtaBins = cms.int32(4),
0012 trainings = cms.VPSet(
0013 cms.PSet(
0014 jEtaMin = cms.double(0.),
0015 jEtaMax = cms.double(2.5),
0016 tmvaWeights = cms.FileInPath("RecoJets/JetProducers/data/pileupJetId_80XvarFix_Eta0to2p5_BDT.weights.xml.gz"),
0017 tmvaVariables = cms.vstring(
0018 "nvtx",
0019 "dR2Mean" ,
0020 "nParticles" ,
0021 "nCharged" ,
0022 "majW" ,
0023 "minW",
0024 "frac01" ,
0025 "frac02" ,
0026 "frac03" ,
0027 "frac04" ,
0028 "ptD" ,
0029 "beta" ,
0030 "pull" ,
0031 "jetR" ,
0032 "jetRchg" ,
0033 )
0034 ),
0035 cms.PSet(
0036 jEtaMin = cms.double(2.5),
0037 jEtaMax = cms.double(2.75),
0038 tmvaWeights = cms.FileInPath("RecoJets/JetProducers/data/pileupJetId_80XvarFix_Eta2p5to2p75_BDT.weights.xml.gz"),
0039 tmvaVariables = cms.vstring(
0040 "nvtx",
0041 "dR2Mean" ,
0042 "nParticles" ,
0043 "nCharged" ,
0044 "majW" ,
0045 "minW",
0046 "frac01" ,
0047 "frac02" ,
0048 "frac03" ,
0049 "frac04" ,
0050 "ptD" ,
0051 "beta" ,
0052 "pull" ,
0053 "jetR" ,
0054 "jetRchg" ,
0055 )
0056 ),
0057 cms.PSet(
0058 jEtaMin = cms.double(2.75),
0059 jEtaMax = cms.double(3.),
0060 tmvaWeights = cms.FileInPath("RecoJets/JetProducers/data/pileupJetId_80XvarFix_Eta2p75to3_BDT.weights.xml.gz"),
0061 tmvaVariables = cms.vstring(
0062 "nvtx",
0063 "dR2Mean" ,
0064 "nParticles" ,
0065 "nCharged" ,
0066 "majW" ,
0067 "minW",
0068 "frac01" ,
0069 "frac02" ,
0070 "frac03" ,
0071 "frac04" ,
0072 "ptD" ,
0073 "beta" ,
0074 "pull" ,
0075 "jetR" ,
0076 "jetRchg" ,
0077 )
0078 ),
0079 cms.PSet(
0080 jEtaMin = cms.double(3.),
0081 jEtaMax = cms.double(5.),
0082 tmvaWeights = cms.FileInPath("RecoJets/JetProducers/data/pileupJetId_80XvarFix_Eta3to5_BDT.weights.xml.gz"),
0083 tmvaVariables = cms.vstring(
0084 "nvtx",
0085 "dR2Mean" ,
0086 "nParticles" ,
0087 "majW" ,
0088 "minW",
0089 "frac01" ,
0090 "frac02" ,
0091 "frac03" ,
0092 "frac04" ,
0093 "ptD" ,
0094 "pull" ,
0095 "jetR" ,
0096 )
0097 ),
0098 ),
0099 tmvaSpectators = cms.vstring(
0100 "jetPt" ,
0101 "jetEta" ,
0102 ),
0103 JetIdParams = full_81x_chs_wp,
0104 label = cms.string("full")
0105 )
0106
0107
0108 trainingVariables_102X_Eta0To3 = [
0109 "nvtx" ,
0110 "beta" ,
0111 "dR2Mean" ,
0112 "frac01" ,
0113 "frac02" ,
0114 "frac03" ,
0115 "frac04" ,
0116 "majW" ,
0117 "minW" ,
0118 "jetR" ,
0119 "jetRchg" ,
0120 "nParticles",
0121 "nCharged" ,
0122 "ptD" ,
0123 "pull" ,
0124 ]
0125 trainingVariables_102X_Eta3To5 = list(trainingVariables_102X_Eta0To3)
0126 trainingVariables_102X_Eta3To5.remove('beta')
0127 trainingVariables_102X_Eta3To5.remove('jetRchg')
0128 trainingVariables_102X_Eta3To5.remove('nCharged')
0129
0130
0131 full_102x_chs = full_81x_chs.clone(
0132 JetIdParams = full_102x_chs_wp,
0133 trainings = {0: dict(tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_102X_Eta0p0To2p5_chs_BDT.weights.xml.gz",
0134 tmvaVariables = trainingVariables_102X_Eta0To3),
0135 1: dict(tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_102X_Eta2p5To2p75_chs_BDT.weights.xml.gz",
0136 tmvaVariables = trainingVariables_102X_Eta0To3),
0137 2: dict(tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_102X_Eta2p75To3p0_chs_BDT.weights.xml.gz",
0138 tmvaVariables = trainingVariables_102X_Eta0To3),
0139 3: dict(tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_102X_Eta3p0To5p0_chs_BDT.weights.xml.gz",
0140 tmvaVariables = trainingVariables_102X_Eta3To5)
0141 }
0142 )
0143
0144
0145 full_94x_chs = full_102x_chs.clone(JetIdParams = full_94x_chs_wp)
0146 for train in full_94x_chs.trainings:
0147 train.tmvaWeights = train.tmvaWeights.value().replace("102X", "94X")
0148
0149
0150 full_106x_UL17_chs = full_102x_chs.clone(JetIdParams = full_106x_UL17_chs_wp)
0151 for train in full_106x_UL17_chs.trainings:
0152 train.tmvaWeights = train.tmvaWeights.value().replace("102X", "UL17")
0153
0154
0155 full_106x_UL18_chs = full_106x_UL17_chs.clone(JetIdParams = full_106x_UL18_chs_wp)
0156 for train in full_106x_UL18_chs.trainings:
0157 train.tmvaWeights = train.tmvaWeights.value().replace("UL17", "UL18")
0158
0159
0160 full_106x_UL16_chs = full_106x_UL17_chs.clone(JetIdParams = full_106x_UL16_chs_wp)
0161 for train in full_106x_UL16_chs.trainings:
0162 train.tmvaWeights = train.tmvaWeights.value().replace("UL17", "UL16")
0163
0164
0165 full_106x_UL16APV_chs = full_106x_UL17_chs.clone(JetIdParams = full_106x_UL16APV_chs_wp)
0166 for train in full_106x_UL16APV_chs.trainings:
0167 train.tmvaWeights = train.tmvaWeights.value().replace("UL17", "UL16APV")
0168
0169
0170 full_133x_Winter24_puppi_v18_wp = full_106x_UL17_chs.clone(
0171 JetIdParams = full_133x_Winter24_puppiv18_wp,
0172 trainings = {0: dict(tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_133X_Winter24_Eta0p0To2p5_puppiV18_BDT.weights.xml.gz",
0173 tmvaVariables = trainingVariables_102X_Eta0To3),
0174 1: dict(tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_133X_Winter24_Eta2p5To2p75_puppiV18_BDT.weights.xml.gz",
0175 tmvaVariables = trainingVariables_102X_Eta0To3),
0176 2: dict(tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_133X_Winter24_Eta2p75To3p0_puppiV18_BDT.weights.xml.gz",
0177 tmvaVariables = trainingVariables_102X_Eta0To3),
0178 3: dict(tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_133X_Winter24_Eta3p0To5p0_puppiV18_BDT.weights.xml.gz",
0179 tmvaVariables = trainingVariables_102X_Eta3To5)
0180 }
0181 )
0182
0183
0184
0185 cutbased = cms.PSet(
0186 impactParTkThreshold = cms.double(1.),
0187 cutBased = cms.bool(True),
0188 JetIdParams = EmptyCutBased_wp,
0189 label = cms.string("cutbased")
0190 )
