python/Identification/mvaElectronID_Summer17UL_ID_ISO_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 from RecoEgamma.ElectronIdentification.Identification.mvaElectronID_tools import *
0003 from os import path
0004
0005 mvaTag = "Summer17ULIdIso"
0006
0007 weightFileDir = "RecoEgamma/ElectronIdentification/data/MVAWeightFiles/Summer_17UL_ID_ISO"
0008
0009 mvaWeightFiles = cms.vstring(
0010      path.join(weightFileDir, "EB1_5.weights.root"), # EB1_5
0011      path.join(weightFileDir, "EB2_5.weights.root"), # EB2_5
0012      path.join(weightFileDir, "EE_5.weights.root"), # EE_5
0013      path.join(weightFileDir, "EB1_10.weights.root"), # EB1_10
0014      path.join(weightFileDir, "EB2_10.weights.root"), # EB2_10
0015      path.join(weightFileDir, "EE_10.weights.root"), # EE_10
0016      )
0017
0018 categoryCuts = cms.vstring(
0019      "pt < 10. & abs(superCluster.eta) < 0.800", # EB1_5
0020      "pt < 10. & abs(superCluster.eta) >= 0.800 & abs(superCluster.eta) < 1.479", # EB2_5
0021      "pt < 10. & abs(superCluster.eta) >= 1.479", # EE_5
0022      "pt >= 10. & abs(superCluster.eta) < 0.800", # EB1_10
0023      "pt >= 10. & abs(superCluster.eta) >= 0.800 & abs(superCluster.eta) < 1.479", # EB2_10
0024      "pt >= 10. & abs(superCluster.eta) >= 1.479", # EE_10
0025      )
0026
0027 mvaEleID_Summer17UL_ID_ISO_HZZ_container = EleMVARaw_WP(
0028     idName = "mvaEleID-Summer17UL-ID-ISO-HZZ", mvaTag = mvaTag,
0029     cutCategory0 = "1.54440585808", # EB1_5
0030     cutCategory1 = "1.50294621563", # EB2_5
0031     cutCategory2 = "1.77306202112", # EE_5
0032     cutCategory3 = "0.157262554087", # EB1_10
0033     cutCategory4 = "0.0273932225081", # EB2_10
0034     cutCategory5 = "-0.623050463489", # EE_10
0035     )
0036
0037
0038 mvaEleID_Summer17UL_ID_ISO_producer_config = cms.PSet(
0039     mvaName             = cms.string(mvaClassName),
0040     mvaTag              = cms.string(mvaTag),
0041     nCategories         = cms.int32(6),
0042     categoryCuts        = categoryCuts,
0043     weightFileNames     = mvaWeightFiles,
0044     variableDefinition  = cms.string(mvaVariablesFile)
0045     )
0046
0047 mvaEleID_Summer17UL_ID_ISO_HZZ = configureVIDMVAEleID( mvaEleID_Summer17UL_ID_ISO_HZZ_container )
0048
0049 mvaEleID_Summer17UL_ID_ISO_HZZ.isPOGApproved = cms.untracked.bool(True)