Merge pull request #32954 from singh-ramanpreet/PUID_UL18_backport_11…

…_2_X PUID UL18 backport to 11_2_X
cms-sw · Feb 25, 2021 · 33644d2 · 33644d2
2 parents 512d558 + 99442f1
commit 33644d2
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/RecoJets/JetProducers/python/PileupJetIDCutParams_cfi.py b/RecoJets/JetProducers/python/PileupJetIDCutParams_cfi.py
@@ -67,6 +67,11 @@
     Pt4050_Loose = cms.vdouble(-0.19, 0.22, -0.13, -0.03)
 )
 
+##########################################################
+## Working points for the 106X UL18 training
+###########################################################
+full_106x_UL18_chs_wp = full_106x_UL17_chs_wp.clone()
+
 ###########################################################
 ## Working points for the 80X training
 ###########################################################

diff --git a/RecoJets/JetProducers/python/PileupJetIDParams_cfi.py b/RecoJets/JetProducers/python/PileupJetIDParams_cfi.py
@@ -163,6 +163,11 @@
 full_106x_UL17_chs.trainings[3].tmvaWeights = "RecoJets/JetProducers/data/pileupJetId_UL17_Eta3p0To5p0_chs_BDT.weights.xml.gz"
 full_106x_UL17_chs.trainings[3].tmvaVariables = trainingVariables_106X_Eta3To5
 
+####################################################################################################################
+full_106x_UL18_chs = full_106x_UL17_chs.clone(JetIdParams = full_106x_UL18_chs_wp)
+for train in full_106x_UL18_chs.trainings:
+    train.tmvaWeights = train.tmvaWeights.value().replace("UL17", "UL18")
+
 ####################################################################################################################
 full_80x_chs = cms.PSet(
         impactParTkThreshold = cms.double(1.),

diff --git a/RecoJets/JetProducers/python/PileupJetID_cfi.py b/RecoJets/JetProducers/python/PileupJetID_cfi.py
@@ -1,5 +1,7 @@
 import FWCore.ParameterSet.Config as cms
-
+from Configuration.Eras.Modifier_run2_miniAOD_devel_cff import run2_miniAOD_devel
+from Configuration.Eras.Modifier_run2_jme_2017_cff import run2_jme_2017
+from Configuration.Eras.Modifier_run2_jme_2016_cff import run2_jme_2016
 from RecoJets.JetProducers.PileupJetIDParams_cfi import *
 
 #_stdalgos_4x = cms.VPSet(full,   cutbased,PhilV1)
@@ -14,6 +16,7 @@
 _chsalgos_94x = cms.VPSet(full_94x_chs,cutbased)
 _chsalgos_102x = cms.VPSet(full_102x_chs,cutbased)
 _chsalgos_106X_UL17 = cms.VPSet(full_106x_UL17_chs,cutbased)
+_chsalgos_106X_UL18 = cms.VPSet(full_106x_UL18_chs,cutbased)
 
 _stdalgos    = _chsalgos_106X_UL17
 
@@ -33,6 +36,7 @@
      usePuppi = cms.bool(False),
 #     residualsTxt     = cms.FileInPath("RecoJets/JetProducers/data/download.url") # must be an existing file
 )
+(run2_miniAOD_devel & (~run2_jme_2016) & (~run2_jme_2017)).toModify(pileupJetId, algos = _chsalgos_106X_UL18)
 
 # Calculate variables, but don't run MVAs
 pileupJetIdCalculator = pileupJetId.clone(