Project CMSSW displayed by LXR CMSSW_13_0_X_2023-01-30-2300/​RecoEgamma/​ElectronIdentification/​python/​Identification/​mvaElectronID_tools.py	Source navigation Diff markup Identifier search General search
	Version: CMSSW_13_0_X_2023-01-30-2300 CMSSW_13_0_X_2023-01-29-2300 CMSSW_13_0_X_2023-01-27-2300 CMSSW_13_0_X_2023-01-26-2300 CMSSW_13_0_X_2023-01-25-2300 CMSSW_13_0_X_2023-01-24-2300 CMSSW_11_2_1 CMSSW_11_1_7 CMSSW_11_0_3 CMSSW_10_6_20 CMSSW_7_6_7 CMSSW_7_5_8_patch3 CMSSW_5_3_32 CMSSW_4_4_7 CMSSW_4_2_8_lowpupatch1 CMSSW_3_9_2 Revert   Change

File indexing completed on 2022-12-09 23:46:49

 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"
 
 # =======================================
 # 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