Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2022-06-14 02:53:35

0001 import FWCore.ParameterSet.Config as cms
0002 from RecoEgamma.ElectronIdentification.Identification.mvaElectronID_tools import *
0003 from os import path
0004 
0005 # Egamma presentation on this ID:
0006 # https://indico.cern.ch/event/732971/contributions/3022864/attachments/1658765/2656595/180530_egamma.pdf
0007 
0008 mvaTag = "Fall17NoIsoV2"
0009 
0010 weightFileDir = "RecoEgamma/ElectronIdentification/data/MVAWeightFiles/Fall17NoIsoV2"
0011 
0012 mvaWeightFiles = cms.vstring(
0013      path.join(weightFileDir, "EB1_5.weights.root"), # EB1_5
0014      path.join(weightFileDir, "EB2_5.weights.root"), # EB2_5
0015      path.join(weightFileDir, "EE_5.weights.root"), # EE_5
0016      path.join(weightFileDir, "EB1_10.weights.root"), # EB1_10
0017      path.join(weightFileDir, "EB2_10.weights.root"), # EB2_10
0018      path.join(weightFileDir, "EE_10.weights.root"), # EE_10
0019      )
0020 
0021 mvaEleID_Fall17_noIso_V2_wp80_container = EleMVARaw_WP(
0022     idName = "mvaEleID-Fall17-noIso-V2-wp80", mvaTag = mvaTag,
0023     cutCategory0 = "3.26449620468 - exp(-pt / 3.32657149223) * 8.84669783568", # EB1_5
0024     cutCategory1 = "2.83557838497 - exp(-pt / 2.15150487651) * 11.0978016567", # EB2_5
0025     cutCategory2 = "2.91994945177 - exp(-pt / 1.69875477522) * 24.024807824", # EE_5
0026     cutCategory3 = "7.1336238874 - exp(-pt / 16.5605268797) * 8.22531222391", # EB1_10
0027     cutCategory4 = "6.18638275782 - exp(-pt / 15.2694634284) * 7.49764565324", # EB2_10
0028     cutCategory5 = "5.43175865738 - exp(-pt / 15.4290075949) * 7.56899692285", # EE_10
0029     )
0030 
0031 mvaEleID_Fall17_noIso_V2_wpLoose_container = EleMVARaw_WP(
0032     idName = "mvaEleID-Fall17-noIso-V2-wpLoose", mvaTag = mvaTag,
0033     cutCategory0 = "0.894411158628", # EB1_5
0034     cutCategory1 = "0.791966464633", # EB2_5
0035     cutCategory2 = "1.47104857173", # EE_5
0036     cutCategory3 = "-0.293962958665", # EB1_10
0037     cutCategory4 = "-0.250424758584", # EB2_10
0038     cutCategory5 = "-0.130985179031", # EE_10
0039     )
0040 
0041 mvaEleID_Fall17_noIso_V2_wp90_container = EleMVARaw_WP(
0042     idName = "mvaEleID-Fall17-noIso-V2-wp90", mvaTag = mvaTag,
0043     cutCategory0 = "2.77072387339 - exp(-pt / 3.81500912145) * 8.16304860178", # EB1_5
0044     cutCategory1 = "1.85602317813 - exp(-pt / 2.18697654938) * 11.8568936824", # EB2_5
0045     cutCategory2 = "1.73489307814 - exp(-pt / 2.0163211971) * 17.013880078", # EE_5
0046     cutCategory3 = "5.9175992258 - exp(-pt / 13.4807294538) * 9.31966232685", # EB1_10
0047     cutCategory4 = "5.01598837255 - exp(-pt / 13.1280451502) * 8.79418193765", # EB2_10
0048     cutCategory5 = "4.16921343208 - exp(-pt / 13.2017224621) * 9.00720913211", # EE_10
0049     )
0050 
0051 workingPoints = dict(
0052     wp80 = mvaEleID_Fall17_noIso_V2_wp80_container,
0053     wpLoose = mvaEleID_Fall17_noIso_V2_wpLoose_container,
0054     wp90 = mvaEleID_Fall17_noIso_V2_wp90_container
0055 )
0056 
0057 mvaEleID_Fall17_noIso_V2_producer_config = cms.PSet(
0058     mvaName             = cms.string(mvaClassName),
0059     mvaTag              = cms.string(mvaTag),
0060     nCategories         = cms.int32(6),
0061     categoryCuts        = cms.vstring(*EleMVA_6CategoriesCuts),
0062     weightFileNames     = mvaWeightFiles,
0063     variableDefinition  = cms.string(mvaVariablesFile)
0064     )
0065 
0066 mvaEleID_Fall17_noIso_V2_wp80 = configureVIDMVAEleID( mvaEleID_Fall17_noIso_V2_wp80_container )
0067 mvaEleID_Fall17_noIso_V2_wpLoose = configureVIDMVAEleID( mvaEleID_Fall17_noIso_V2_wpLoose_container )
0068 mvaEleID_Fall17_noIso_V2_wp90 = configureVIDMVAEleID( mvaEleID_Fall17_noIso_V2_wp90_container )
0069 
0070 mvaEleID_Fall17_noIso_V2_wp80.isPOGApproved = cms.untracked.bool(True)
0071 mvaEleID_Fall17_noIso_V2_wpLoose.isPOGApproved = cms.untracked.bool(True)
0072 mvaEleID_Fall17_noIso_V2_wp90.isPOGApproved = cms.untracked.bool(True)