Backporting DeepDoubleX(BvL,CvL,CvB) tagger to 94X

cms-sw · Nov 29, 2018 · 8db949c · 8db949c
1 parent bf15e91
commit 8db949c
Show file tree

Hide file tree

Showing 9 changed files with 15 additions and 17 deletions.
diff --git a/PhysicsTools/PatAlgos/python/slimming/applyDeepBtagging_cff.py b/PhysicsTools/PatAlgos/python/slimming/applyDeepBtagging_cff.py
@@ -42,7 +42,6 @@ def applyDeepBtagging( process, postfix="" ) :
     # delete module not used anymore (slimmedJets substitutes)
     delattr(process, 'selectedUpdatedPatJetsSlimmedDeepFlavour'+postfix)
 
-<<<<<<< HEAD
     #switch to True if needed for private tests
     enableDeepTags = False
     if enableDeepTags:

diff --git a/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDB.pb b/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDB.pb
diff --git a/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDB_mass_independent.pb b/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDB_mass_independent.pb
diff --git a/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDC.pb b/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDC.pb
diff --git a/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDC_mass_independent.pb b/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDC_mass_independent.pb
diff --git a/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDCvB.pb b/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDCvB.pb
diff --git a/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDCvB_mass_independent.pb b/RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDCvB_mass_independent.pb
diff --git a/RecoBTag/TensorFlow/python/pfMassIndependentDeepDoubleXJetTags_cff.py b/RecoBTag/TensorFlow/python/pfMassIndependentDeepDoubleXJetTags_cff.py
@@ -4,12 +4,10 @@
 from pfDeepDoubleCvBJetTags_cfi import pfDeepDoubleCvBJetTags
 
 pfMassIndependentDeepDoubleBvLJetTags = pfDeepDoubleBvLJetTags.clone(
-    graph_path = cms.FileInPath('RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDB_mass_independent.pb')
-)
+    graph_path = 'RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDB_mass_independent.pb')
 pfMassIndependentDeepDoubleCvLJetTags = pfDeepDoubleCvLJetTags.clone(
-    graph_path = cms.FileInPath('RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDC_mass_independent.pb')
-)
+    graph_path = 'RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDC_mass_independent.pb')
 pfMassIndependentDeepDoubleCvBJetTags = pfDeepDoubleCvBJetTags.clone(
-    graph_path = cms.FileInPath('RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDCvB_mass_independent.pb')
-)
+    graph_path = 'RecoBTag/Combined/data/DeepDoubleX/94X/V01/DDCvB_mass_independent.pb')
+
 
diff --git a/RecoBTag/TensorFlow/test/plotDDX.py b/RecoBTag/TensorFlow/test/plotDDX.py
@@ -1,3 +1,4 @@
+from __future__ import print_function
 import ROOT
 from DataFormats.FWLite import Handle, Events
 
@@ -16,16 +17,16 @@
     event.getByLabel (labelJ, handleJ)
     jets = handleJ.product()
     for jet in jets  :
-	if jet.pt() < 300 or jet.pt() > 2000: continue
-	if jet.mass() < 40 or jet.mass() > 200: continue
-
-        print jet.pt(), jet.mass()
-        print "DDB", jet.bDiscriminator("pfDeepDoubleBvLJetTags:probQCD"), jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb") , jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHcc")
-        print "DDB", jet.bDiscriminator("pfMassIndependentDeepDoubleBvLJetTags:probQCD"), jet.bDiscriminator("pfMassIndependentDeepDoubleBvLJetTags:probHbb") , jet.bDiscriminator("pfMassIndependentDeepDoubleBvLJetTags:probHcc")
-        print "DDCvL", jet.bDiscriminator("pfDeepDoubleCvLJetTags:probQCD"), jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHcc") , jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHbb")
-        print "DDCvL", jet.bDiscriminator("pfMassIndependentDeepDoubleCvLJetTags:probQCD"), jet.bDiscriminator("pfMassIndependentDeepDoubleCvLJetTags:probHcc") , jet.bDiscriminator("pfDeepDoubleMassIndependentCvLJetTags:probHbb")
-        print "DDCvB", jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHbb"), jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHcc") , jet.bDiscriminator("pfDeepDoubleCvBJetTags:probQCD")
-        print "DDCvB", jet.bDiscriminator("pfMassIndependentDeepDoubleCvBJetTags:probHbb"), jet.bDiscriminator("pfMassIndependentDeepDoubleCvBJetTags:probHcc") , jet.bDiscriminator("pfMassIndependentDeepDoubleCvBJetTags:probQCD")
+        if jet.pt() < 300 or jet.pt() > 2000: continue
+        if jet.mass() < 40 or jet.mass() > 200: continue
+
+        print(jet.pt(), jet.mass())
+        print("DDB", jet.bDiscriminator("pfDeepDoubleBvLJetTags:probQCD"), jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb"))
+        print("DDB", jet.bDiscriminator("pfMassIndependentDeepDoubleBvLJetTags:probQCD"), jet.bDiscriminator("pfMassIndependentDeepDoubleBvLJetTags:probHbb"))
+        print("DDCvL", jet.bDiscriminator("pfDeepDoubleCvLJetTags:probQCD"), jet.bDiscriminator("pfDeepDoubleCvLJetTags:probHcc"))
+        print("DDCvL", jet.bDiscriminator("pfMassIndependentDeepDoubleCvLJetTags:probQCD"), jet.bDiscriminator("pfMassIndependentDeepDoubleCvLJetTags:probHcc"))
+        print("DDCvB", jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHbb"), jet.bDiscriminator("pfDeepDoubleCvBJetTags:probHcc") )
+        print("DDCvB", jet.bDiscriminator("pfMassIndependentDeepDoubleCvBJetTags:probHbb"), jet.bDiscriminator("pfMassIndependentDeepDoubleCvBJetTags:probHcc"))
         h_probQ_ddb.Fill(jet.bDiscriminator("pfDeepDoubleBvLJetTags:probQCD"))
         h_probH_ddb.Fill(jet.bDiscriminator("pfDeepDoubleBvLJetTags:probHbb"))
         h_probQ_ddc.Fill(jet.bDiscriminator("pfDeepDoubleCvLJetTags:probQCD"))