File indexing completed on 2023-10-25 10:02:24
0001
0002
0003 import RecoTracker.IterativeTracking.iterativeTkConfig as _cfg
0004 import RecoTracker.IterativeTracking.iterativeTk_cff as _iterativeTk_cff
0005
0006 def getMVASelectors(postfix):
0007
0008
0009
0010 ret = {}
0011
0012 for iterName, seqName in _cfg.iterationAlgos(postfix, includeSequenceName=True):
0013 if hasattr(_iterativeTk_cff, iterName):
0014 mod = getattr(_iterativeTk_cff, iterName)
0015 seq = getattr(_iterativeTk_cff, seqName)
0016
0017
0018 if not seq.contains(mod):
0019 continue
0020
0021 typeName = mod._TypedParameterizable__type
0022 classifiers = []
0023 if typeName == "ClassifierMerger":
0024 classifiers = mod.inputClassifiers.value()
0025 elif "TrackMVAClassifier" in typeName or "TrackLwtnnClassifier" or "TrackTfClassifier" in typeName:
0026 classifiers = [iterName]
0027 if len(classifiers) > 0:
0028 ret[iterName] = (iterName+"Tracks", classifiers)
0029
0030 return ret