mvaPhotonID_tools Namespace Reference

Classes
class	PhoMVA_2Categories_WP

Functions
def	configureFullVIDMVAPhoID
def	configureVIDMVAPhoID_V1

Variables
tuple	category_cuts
float	ebeeSplit = 1.479
string	mvaClassName = "PhotonMVAEstimator"
string	mvaProducerModuleLabel = "photonMVAValueMapProducer"
string	weightFileBaseDir = "RecoEgamma/PhotonIdentification/data/MVA"

Function Documentation

def mvaPhotonID_tools.configureFullVIDMVAPhoID	(		mvaTag,
			variablesFile,
			weightFiles,
			wpConfig,
			addKwargsForValueProducer
	)

Definition at line 93 of file mvaPhotonID_tools.py.

References configureVIDMVAPhoID_V1().

def configureFullVIDMVAPhoID(mvaTag, variablesFile, weightFiles, wpConfig, **addKwargsForValueProducer):

    mvaValueMapName        = mvaProducerModuleLabel + ":" + mvaClassName + mvaTag + "Values"
    mvaCategoriesMapName   = mvaProducerModuleLabel + ":" + mvaClassName + mvaTag + "Categories"

    # Create the PSet that will be fed to the MVA value map producer
    producer_config = cms.PSet( 
        mvaName             = cms.string(mvaClassName),
        mvaTag              = cms.string(mvaTag),
        weightFileNames     = cms.vstring(*weightFiles),
        variableDefinition  = cms.string(variablesFile),
        **addKwargsForValueProducer
        )

    # Create the VPset's for VID cuts
    VID_config = {}
    for wpc in wpConfig:
        idName = wpc["idName"]
        VID_config[idName] = configureVIDMVAPhoID_V1(
                PhoMVA_2Categories_WP(
                    idName = idName,
                    mvaValueMapName = mvaValueMapName,           # map with MVA values for all particles
                    mvaCategoriesMapName = mvaCategoriesMapName, # map with category index for all particles
                    cutCategory0 =  wpc["cuts"]["EB"],  # EB
                    cutCategory1 =  wpc["cuts"]["EE"]   # EE
                )
            )

    configs = {"producer_config": producer_config, "VID_config": VID_config}
    return configs

def mvaPhotonID_tools.configureVIDMVAPhoID_V1

(

mvaWP

)

This function configures the full cms.PSet for a VID ID and returns it.
The inputs: an object of the class PhoMVA_2Categories_WP or similar
that contains all necessary parameters for this MVA.

Definition at line 54 of file mvaPhotonID_tools.py.

Referenced by configureFullVIDMVAPhoID().

def configureVIDMVAPhoID_V1( mvaWP ):
    """
    This function configures the full cms.PSet for a VID ID and returns it.
    The inputs: an object of the class PhoMVA_2Categories_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("PhoMVACut"),
                      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

# ===============================================
# High level function to create a two category ID
# ===============================================

# mvaTag:
#         The mvaTag is an extra string attached to the names of the products
#         such as ValueMaps that needs to distinguish cases when the same MVA estimator
#         class is used with different tuning/weights
# variablesFile:
#         The file listing the variables used in this MVA
# weightFiles:
#         The weight files in the order EB first, then EE
# wpConfig:
#         A dictionary with the names and cut values of the working points
# addKwargsForValueProducer:
#         Additional keyword parameters passed to the producer config