RecoTracker/FinalTrackSelectors/python/trackSelectionTf_cfi.py - master - githubmirror/cmssw

RecoTracker/FinalTrackSelectors/python/trackSelectionTf_cfi.py

13 строк739 B

Andrew Hart

Use model trained using only DisplacedRegionalStep tracks.

20 ноя 2023, 13:43

20 ноя 2023, 13:43d5a94bc

from PhysicsTools.TensorFlow.tfGraphDefProducer_cfi import tfGraphDefProducer as _tfGraphDefProducer
trackSelectionTf = _tfGraphDefProducer.clone(
    ComponentName = "trackSelectionTf",
    FileName = "RecoTracker/FinalTrackSelectors/data/TrackTfClassifier/MkFit_Run3_12_5_0_pre5.pb"
)
trackSelectionTfPLess = _tfGraphDefProducer.clone(
    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"
)