RecoTracker/IterativeTracking/python/iterativeTkUtils.py - master - githubmirror/cmssw

RecoTracker/IterativeTracking/python/iterativeTkUtils.py

30 строк1 KB

hajohajo@helsinki.fi

Add track classifier that uses tensorflow.

07 фев 2021, 07:24

07 фев 2021, 07:244e13557

# This file provides additional helpers for getting information of
# iterations in automated way.
import RecoTracker.IterativeTracking.iterativeTkConfig as _cfg
import RecoTracker.IterativeTracking.iterativeTk_cff as _iterativeTk_cff

def getMVASelectors(postfix):
    # assume naming convention that the iteration name (when first
    # letter in lower case) is the selector name

    ret = {}

    for iterName, seqName in _cfg.iterationAlgos(postfix, includeSequenceName=True):
        if hasattr(_iterativeTk_cff, iterName):
            mod = getattr(_iterativeTk_cff, iterName)
            seq = getattr(_iterativeTk_cff, seqName)

            # Ignore iteration if the MVA selector module is not in the sequence
            if not seq.contains(mod):
                continue

            typeName = mod._TypedParameterizable__type
            classifiers = []
            if typeName == "ClassifierMerger":
                classifiers = mod.inputClassifiers.value()
            elif "TrackMVAClassifier" in typeName or "TrackLwtnnClassifier" or "TrackTfClassifier" in typeName:
                classifiers = [iterName]
            if len(classifiers) > 0:
                ret[iterName] = (iterName+"Tracks", classifiers)

    return ret