Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2024-04-06 12:25:08

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 for Run3
0006 # https://indico.cern.ch/event/1220628/contributions/5134878/attachments/2546114/4384580/Run%203%20Electron%20MVA%20based%20ID%20training.pdf
0007 
0008 mvaTag = "RunIIIWinter22IsoV1"
0009 
0010 weightFileDir = "RecoEgamma/ElectronIdentification/data/MVAWeightFiles/Winter22IsoV1"
0011 
0012 mvaWeightFiles = [
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_RunIIIWinter22_iso_V1_wp80_container = EleMVA_WP(
0022     idName = "mvaEleID-RunIIIWinter22-iso-V1-wp80", mvaTag = mvaTag,
0023     cutCategory0 = "0.99776583820026143", # EB1_5
0024     cutCategory1 = "0.99399710641666705", # EB2_5
0025     cutCategory2 = "0.97627554114737425", # EE_5
0026     cutCategory3 = "0.99997133733482069", # EB1_10
0027     cutCategory4 = "0.99991566148661426", # EB2_10
0028     cutCategory5 = "0.99932288865775143", # EE_10
0029     )
0030 
0031 
0032 mvaEleID_RunIIIWinter22_iso_V1_wp90_container = EleMVA_WP(
0033     idName = "mvaEleID-RunIIIWinter22-iso-V1-wp90", mvaTag = mvaTag,
0034     cutCategory0 = "0.9870981346957135", # EB1_5
0035     cutCategory1 = "0.95756807831082225", # EB2_5
0036     cutCategory2 = "0.81079745560696059", # EE_5
0037     cutCategory3 = "0.99981763428587134", # EB1_10
0038     cutCategory4 = "0.99936974968805936", # EB2_10
0039     cutCategory5 = "0.99293135832977431", # EE_10
0040     )
0041 
0042 
0043 workingPoints = dict(
0044     wp80 = mvaEleID_RunIIIWinter22_iso_V1_wp80_container,
0045     wp90 = mvaEleID_RunIIIWinter22_iso_V1_wp90_container
0046 )
0047 
0048 mvaEleID_RunIIIWinter22_iso_V1_producer_config = cms.PSet(
0049     mvaName             = cms.string(mvaClassName),
0050     mvaTag              = cms.string(mvaTag),
0051     nCategories         = cms.int32(6),
0052     categoryCuts        = cms.vstring(*EleMVA_6CategoriesCuts),
0053     weightFileNames     = cms.vstring(*mvaWeightFiles),
0054     variableDefinition  = cms.string(mvaVariablesFileRun3)
0055     )
0056 
0057 
0058 mvaEleID_RunIIIWinter22_iso_V1_wp80 = configureVIDMVAEleID( mvaEleID_RunIIIWinter22_iso_V1_wp80_container )
0059 mvaEleID_RunIIIWinter22_iso_V1_wp90 = configureVIDMVAEleID( mvaEleID_RunIIIWinter22_iso_V1_wp90_container )
0060 
0061 
0062 mvaEleID_RunIIIWinter22_iso_V1_wp80.isPOGApproved = cms.untracked.bool(True)
0063 mvaEleID_RunIIIWinter22_iso_V1_wp90.isPOGApproved = cms.untracked.bool(True)