Project CMSSW displayed by LXR CMSSW_15_1_X_2025-07-14-2300/​RecoEgamma/​ElectronIdentification/​python/​Identification/​mvaElectronID_tools.py	Source navigation Diff markup Identifier search General search
	Version: CMSSW_15_1_X_2025-07-14-2300 CMSSW_15_1_X_2025-07-13-2300 CMSSW_15_1_X_2025-07-11-2300 CMSSW_15_1_X_2025-07-10-2300 CMSSW_15_1_X_2025-07-08-2300 CMSSW_15_1_X_2025-07-07-2300 CMSSW_15_0_4 CMSSW_15_0_3_patch1 CMSSW_14_0_21_patch1 Revert   Change

File indexing completed on 2024-04-06 12:25:08

 import FWCore.ParameterSet.Config as cms
 
 # =========================
 # Various commom parameters
 # =========================
 
 # This MVA implementation class name
 mvaClassName = "ElectronMVAEstimatorRun2"
 
 # The locatoins of value maps with the actual MVA values and categories
 # for all particles.
 # The names for the maps are "<module name>:<MVA class name>Values"
 # and "<module name>:<MVA class name>Categories"
 mvaProducerModuleLabel = "electronMVAValueMapProducer"
 
 # The files with the variable definitions
 mvaVariablesFile            = "RecoEgamma/ElectronIdentification/data/ElectronMVAEstimatorRun2Variables.txt"
 mvaVariablesFileRun3     = "RecoEgamma/ElectronIdentification/data/ElectronIDVariablesRun3.txt"
 mvaVariablesFileRun3NonIso  = "RecoEgamma/ElectronIdentification/data/ElectronIDVariablesRun3NonIso.txt"
 
 # =======================================
 # Define some commonly used category cuts
 # =======================================
 
 EleMVA_3CategoriesCuts = [
     "abs(superCluster.eta) < 0.800",
     "abs(superCluster.eta) >= 0.800 && abs(superCluster.eta) < 1.479",
     "abs(superCluster.eta) >= 1.479"
     ]
 
 EleMVA_6CategoriesCuts = [
     "pt < 10. && abs(superCluster.eta) < 0.800",
     "pt < 10. && abs(superCluster.eta) >= 0.800 && abs(superCluster.eta) < 1.479",
     "pt < 10. && abs(superCluster.eta) >= 1.479",
     "pt >= 10. && abs(superCluster.eta) < 0.800",
     "pt >= 10. && abs(superCluster.eta) >= 0.800 && abs(superCluster.eta) < 1.479",
     "pt >= 10. && abs(superCluster.eta) >= 1.479",
     ]
 
 # =======================================================
 # Define simple containers for MVA cut values and related
 # =======================================================
 
 class EleMVA_WP:
     """
     This is a container class to hold MVA cut values for a n-category MVA
     as well as the names of the value maps that contain the MVA values computed
     for all particles in a producer upstream.
 
     IMPORTANT: the cuts need to be given in alphabetical order, which must
     be the order in which they are used by the cut class.
     """
     def __init__(self,
                  idName,
                  mvaTag,
                  **cuts
                  ):
         self.idName               = idName
         # map with MVA values for all particles
         self.mvaValueMapName      = mvaProducerModuleLabel + ":" + mvaClassName + mvaTag + "Values"
         # map with category index for all particles
         self.mvaCategoriesMapName = mvaProducerModuleLabel + ":" + mvaClassName + mvaTag + "Categories"
         self.cuts = cuts
 
     def getCutStrings(self):
         keylist = sorted(self.cuts.keys())
         return [self.cuts[key] for key in keylist]
 
 class EleMVARaw_WP:
     """
     This is a container class to hold MVA cut values for a n-category MVA
     as well as the names of the value maps that contain the MVA values computed
     for all particles in a producer upstream.
 
     IMPORTANT: the cuts need to be given in alphabetical order, which must
     be the order in which they are used by the cut class.
     """
     def __init__(self,
                  idName,
                  mvaTag,
                  **cuts
                  ):
         self.idName               = idName
         # map with MVA values for all particles
         self.mvaValueMapName      = mvaProducerModuleLabel + ":" + mvaClassName + mvaTag + "RawValues"
         # map with category index for all particles
         self.mvaCategoriesMapName = mvaProducerModuleLabel + ":" + mvaClassName + mvaTag + "Categories"
         self.cuts = cuts
 
     def getCutStrings(self):
         keylist = sorted(self.cuts.keys())
         return [self.cuts[key] for key in keylist]
 
 # ================================
 # Define the complete MVA cut sets
 # ================================
 
 def configureVIDMVAEleID(mvaWP, cutName="GsfEleMVACut"):
     """
     This function configures the full cms.PSet for a VID ID and returns it.
     The inputs: an object of the class EleMVA_WP or similar
     that contains all necessary parameters for this MVA.
     """
     pSet = cms.PSet(
         #
         idName = cms.string( mvaWP.idName ),
         cutFlow = cms.VPSet(
             cms.PSet( cutName = cms.string(cutName),
                       mvaCuts = cms.vstring( mvaWP.getCutStrings() ),
                       mvaValueMapName = cms.InputTag( mvaWP.mvaValueMapName ),
                       mvaCategoriesMapName = cms.InputTag( mvaWP.mvaCategoriesMapName ),
                       needsAdditionalProducts = cms.bool(True),
                       isIgnored = cms.bool(False)
                       )
             )
         )
     #
     return pSet

This page was automatically generated by the 2.2.1 LXR engine. The LXR team