Merge pull request #43336 from aehart/displaced_regional_dnn

cmsbuild · web-flow · commit df4abb9f4221 · 2023-11-29T19:31:39.000+01:00
DisplacedRegionalStep track DNN
diff --git a/RecoTracker/FinalTrackSelectors/python/trackSelectionTf_cfi.py b/RecoTracker/FinalTrackSelectors/python/trackSelectionTf_cfi.py
@@ -7,5 +7,9 @@
     ComponentName = "trackSelectionTfPLess",
     FileName = "RecoTracker/FinalTrackSelectors/data/TrackTfClassifier/MkFitPixelLessOnly_Run3_12_5_0_pre5.pb"
 )
+trackSelectionTfDisplacedRegional = _tfGraphDefProducer.clone(
+    ComponentName = "trackSelectionTfDisplacedRegional",
+    FileName = "RecoTracker/FinalTrackSelectors/data/TrackTfClassifier/MkFitDisplacedRegionalOnly_Run3_12_5_0_pre5.pb"
+)
 
 
diff --git a/RecoTracker/IterativeTracking/python/DisplacedRegionalStep_cff.py b/RecoTracker/IterativeTracking/python/DisplacedRegionalStep_cff.py
@@ -453,6 +453,7 @@
 from RecoTracker.FinalTrackSelectors.trackSelectionTf_cfi import *
 from RecoTracker.FinalTrackSelectors.trackSelectionTf_CKF_cfi import *
 trackdnn.toReplaceWith(displacedRegionalStep, trackTfClassifier.clone(
+     mva         = dict(tfDnnLabel  = 'trackSelectionTfDisplacedRegional'),
      src         = 'displacedRegionalStepTracks',
      qualityCuts = qualityCutDictionary.DisplacedRegionalStep.value()
 ))

Original file line number	Diff line number	Diff line change
`@@ -7,5 +7,9 @@`
`7`	`7`	`ComponentName = "trackSelectionTfPLess",`
`8`	`8`	`FileName = "RecoTracker/FinalTrackSelectors/data/TrackTfClassifier/MkFitPixelLessOnly_Run3_12_5_0_pre5.pb"`
`9`	`9`	`)`
	`10`	`+trackSelectionTfDisplacedRegional = _tfGraphDefProducer.clone(`
	`11`	`+ ComponentName = "trackSelectionTfDisplacedRegional",`
	`12`	`+ FileName = "RecoTracker/FinalTrackSelectors/data/TrackTfClassifier/MkFitDisplacedRegionalOnly_Run3_12_5_0_pre5.pb"`
	`13`	`+)`
`10`	`14`
`11`	`15`