Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-06-03 02:42:01

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 cutbased = cms.PSet(
0171     impactParTkThreshold = cms.double(1.),
0172     cutBased = cms.bool(True),
0173     JetIdParams = EmptyCutBased_wp,
0174     label = cms.string("cutbased")
0175 )