mvaElectronID_tools.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms

# =======================================================
# Define simple containers for MVA cut values and related
# =======================================================

class EleMVA_3Categories_WP:
    """
    This is a container class to hold MVA cut values for a 3-category MVA
    as well as the names of the value maps that contain the MVA values computed
    for all particles in a producer upstream.
    """
    def __init__(self,
                 idName,
                 mvaValueMapName,
                 mvaCategoriesMapName,
                 cutCategory0,
                 cutCategory1,
                 cutCategory2
                 ):
        self.idName       = idName
        self.mvaValueMapName      = mvaValueMapName
        self.mvaCategoriesMapName = mvaCategoriesMapName
        self.cutCategory0 = cutCategory0
        self.cutCategory1 = cutCategory1
        self.cutCategory2 = cutCategory2

    def getCutValues(self):
        return [self.cutCategory0, self.cutCategory1, self.cutCategory2]

class EleMVA_6Categories_WP:
    """
    This is a container class to hold MVA cut values for a 6-category MVA
    as well as the names of the value maps that contain the MVA values computed
    for all particles in a producer upstream.
    """
    def __init__(self,
                 idName,
                 mvaValueMapName,
                 mvaCategoriesMapName,
                 cutCategory0,
                 cutCategory1,
                 cutCategory2,
                 cutCategory3,
                 cutCategory4,
                 cutCategory5
                 ):
        self.idName       = idName
        self.mvaValueMapName      = mvaValueMapName
        self.mvaCategoriesMapName = mvaCategoriesMapName
        self.cutCategory0 = cutCategory0
        self.cutCategory1 = cutCategory1
        self.cutCategory2 = cutCategory2
        self.cutCategory3 = cutCategory3
        self.cutCategory4 = cutCategory4
        self.cutCategory5 = cutCategory5

    def getCutValues(self):
        return [self.cutCategory0, self.cutCategory1, self.cutCategory2,
                self.cutCategory3, self.cutCategory4, self.cutCategory5]

# ==============================================================
# Define the complete MVA cut sets
# ==============================================================
    
def configureVIDMVAEleID_V1( mvaWP ):
    """
    This function configures the full cms.PSet for a VID ID and returns it.
    The inputs: an object of the class EleMVA_6Categories_WP or similar
    that contains all necessary parameters for this MVA.
    """
    parameterSet =  cms.PSet(
        #
        idName = cms.string( mvaWP.idName ), 
        cutFlow = cms.VPSet( 
            cms.PSet( cutName = cms.string("GsfEleMVACut"),
                      mvaCuts = cms.vdouble( mvaWP.getCutValues() ),
                      mvaValueMapName = cms.InputTag( mvaWP.mvaValueMapName ),
                      mvaCategoriesMapName =cms.InputTag( mvaWP.mvaCategoriesMapName ),
                      needsAdditionalProducts = cms.bool(True),
                      isIgnored = cms.bool(False)
                      )
            )
        )
    #
    return parameterSet