preparation for DeepJet in DQM

cms-sw · Dec 11, 2020 · 18fa48e · 18fa48e
1 parent ec162f0
commit 18fa48e
Show file tree

Hide file tree

Showing 3 changed files with 64 additions and 2 deletions.
diff --git a/DQMOffline/RecoB/python/bTagCommon_cff.py b/DQMOffline/RecoB/python/bTagCommon_cff.py
@@ -84,5 +84,26 @@
             discrCut = cms.double(0.28)
         ),
 
+        cms.PSet(
+            bTagGenericAnalysisBlock,
+            label = cms.InputTag("pfDeepFlavourDiscriminatorsJetTags:BvsAll"),
+            folder = cms.string("deepJet_BvsAll"),
+            differentialPlots = cms.bool(True),
+            discrCut = cms.double(-1)
+        ),
+        #cms.PSet(
+            #bTagGenericAnalysisBlock,
+            #label = cms.InputTag("pfDeepFlavourDiscriminatorsJetTags:CvsL"),
+            #folder = cms.string("deepJet_CvsL"),
+            #differentialPlots = cms.bool(True),
+            #discrCut = cms.double(-1)
+        #),
+        #cms.PSet(
+            #bTagGenericAnalysisBlock,
+            #label = cms.InputTag("pfDeepFlavourDiscriminatorsJetTags:CvsB"),
+            #folder = cms.string("deepJet_CvsB"),
+            #differentialPlots = cms.bool(True),
+            #discrCut = cms.double(-1)
+        #),
     )
 )
diff --git a/DQMOffline/RecoB/python/dqmAnalyzer_cff.py b/DQMOffline/RecoB/python/dqmAnalyzer_cff.py
@@ -1,14 +1,16 @@
 import FWCore.ParameterSet.Config as cms
 
 from RecoBTag.Combined.pfDeepCSVDiscriminatorsJetTags_cfi import pfDeepCSVDiscriminatorsJetTags
+from RecoBTag.Combined.pfDeepFlavourDiscriminatorsJetTags_cfi import pfDeepFlavourDiscriminatorsJetTags
+
 
 ######### DATA ############
 from DQMOffline.RecoB.bTagAnalysisData_cfi import *
 bTagAnalysis.ptRanges = cms.vdouble(0.0)
 bTagAnalysis.doJetID = True
 bTagAnalysis.doJEC = True
 #Residual correction will be added inside the c++ code only for data (checking the presence of genParticles collection), not explicit here as this sequence also ran on MC FullSim
-bTagPlotsDATA = cms.Sequence(pfDeepCSVDiscriminatorsJetTags * bTagAnalysis)
+bTagPlotsDATA = cms.Sequence(pfDeepCSVDiscriminatorsJetTags * pfDeepFlavourDiscriminatorsJetTags * bTagAnalysis)
 
 ## customizations for the pp_on_AA eras
 from Configuration.Eras.Modifier_pp_on_XeXe_2017_cff import pp_on_XeXe_2017
@@ -55,7 +57,7 @@
 bTagValidation.doJEC = True
 bTagValidation.genJetsMatched = cms.InputTag("newpatJetGenJetMatch")
 #to run on fastsim
-prebTagSequenceMC = cms.Sequence(ak4GenJetsForPUid*newpatJetGenJetMatch*selectedHadronsAndPartons*myak4JetFlavourInfos*pfDeepCSVDiscriminatorsJetTags)
+prebTagSequenceMC = cms.Sequence(ak4GenJetsForPUid*newpatJetGenJetMatch*selectedHadronsAndPartons*myak4JetFlavourInfos*pfDeepCSVDiscriminatorsJetTags*pfDeepFlavourDiscriminatorsJetTags)
 bTagPlotsMC = cms.Sequence(bTagValidation)
 
 ## customizations for the pp_on_AA eras

diff --git a/RecoBTag/Combined/python/pfDeepFlavourDiscriminatorsJetTags_cfi.py b/RecoBTag/Combined/python/pfDeepFlavourDiscriminatorsJetTags_cfi.py
@@ -0,0 +1,39 @@
+import FWCore.ParameterSet.Config as cms
+
+pfDeepFlavourDiscriminatorsJetTags = cms.EDProducer(
+   'BTagProbabilityToDiscriminator',
+   discriminators = cms.VPSet(
+      cms.PSet(
+         name = cms.string('BvsAll'),
+         numerator = cms.VInputTag(
+            cms.InputTag('pfDeepFlavourJetTags', 'probb'),
+            cms.InputTag('pfDeepFlavourJetTags', 'probbb'),
+            cms.InputTag('pfDeepFlavourJetTags', 'problepb'),
+            ),
+         denominator = cms.VInputTag(),
+         ),
+      cms.PSet(
+         name = cms.string('CvsB'),
+         numerator = cms.VInputTag(
+            cms.InputTag('pfDeepFlavourJetTags', 'probc'),
+            ),
+         denominator = cms.VInputTag(
+            cms.InputTag('pfDeepFlavourJetTags', 'probc'),
+            cms.InputTag('pfDeepFlavourJetTags', 'probb'),
+            cms.InputTag('pfDeepFlavourJetTags', 'probbb'),
+            cms.InputTag('pfDeepFlavourJetTags', 'problepb'),
+            ),
+         ),
+      cms.PSet(
+         name = cms.string('CvsL'),
+         numerator = cms.VInputTag(
+            cms.InputTag('pfDeepFlavourJetTags', 'probc'),
+            ),
+         denominator = cms.VInputTag(
+            cms.InputTag('pfDeepFlavourJetTags', 'probc'),
+            cms.InputTag('pfDeepFlavourJetTags', 'probuds'),
+            cms.InputTag('pfDeepFlavourJetTags', 'probg'),
+            ),
+         ),
+      )
+   )