python/Identification/mvaElectronID_Fall17_iso_V1_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002 from RecoEgamma.ElectronIdentification.Identification.mvaElectronID_tools import *
0003
0004 # Documentation of the MVA
0005 # https://twiki.cern.ch/twiki/bin/viewauth/CMS/MultivariateElectronIdentificationRun2
0006 # https://rembserj.web.cern.ch/rembserj/notes/Electron_MVA_ID_2017_documentation
0007
0008 #
0009 # In this file we define the locations of the MVA weights, cuts on the MVA values
0010 # for specific working points, and configure those cuts in VID
0011 #
0012
0013 # The tag is an extra string attached to the names of the products
0014 # such as ValueMaps that needs to distinguish cases when the same MVA estimator
0015 # class is used with different tuning/weights
0016 mvaTag = "Fall17IsoV1"
0017
0018 # There are 6 categories in this MVA. They have to be configured in this strict order
0019 # (cuts and weight files order):
0020 #   0   EB1 (eta<0.8)  pt 5-10 GeV     |   pt < ptSplit && |eta| < ebSplit
0021 #   1   EB2 (eta>=0.8) pt 5-10 GeV     |   pt < ptSplit && |eta| >= ebSplit && |eta| < ebeeSplit
0022 #   2   EE             pt 5-10 GeV     |   pt < ptSplit && |eta| >= ebeeSplit
0023 #   3   EB1 (eta<0.8)  pt 10-inf GeV   |   pt >= ptSplit && |eta| < ebSplit
0024 #   4   EB2 (eta>=0.8) pt 10-inf GeV   |   pt >= ptSplit && |eta| >= ebSplit && |eta| < ebeeSplit
0025 #   5   EE             pt 10-inf GeV   |   pt >= ptSplit && |eta| >= ebeeSplit
0026
0027
0028 mvaFall17WeightFiles_V1 = cms.vstring(
0029     "RecoEgamma/ElectronIdentification/data/Fall17/EIDmva_EB1_5_2017_puinfo_iso_BDT.weights.root",
0030     "RecoEgamma/ElectronIdentification/data/Fall17/EIDmva_EB2_5_2017_puinfo_iso_BDT.weights.root",
0031     "RecoEgamma/ElectronIdentification/data/Fall17/EIDmva_EE_5_2017_puinfo_iso_BDT.weights.root",
0032     "RecoEgamma/ElectronIdentification/data/Fall17/EIDmva_EB1_10_2017_puinfo_iso_BDT.weights.root",
0033     "RecoEgamma/ElectronIdentification/data/Fall17/EIDmva_EB2_10_2017_puinfo_iso_BDT.weights.root",
0034     "RecoEgamma/ElectronIdentification/data/Fall17/EIDmva_EE_10_2017_puinfo_iso_BDT.weights.root"
0035     )
0036
0037 ## The working point for this MVA that is expected to have about 90% signal
0038 # WP tuned to give about 90 and 80% signal efficiecny for electrons from Drell-Yan with pT > 25 GeV
0039 # The working point for the low pt categories is just taken over from the high pt
0040 idName90 = "mvaEleID-Fall17-iso-V1-wp90"
0041 MVA_WP90 = EleMVA_WP(
0042     idName90, mvaTag,
0043     cutCategory0 = "0.9387070396095831 - exp(-pt /   2.6525585228167636) *  0.8222647164151365", # EB1 low pt
0044     cutCategory1 = "0.8948802925677235 - exp(-pt /   2.7645670358783523) *  0.4123381218697539", # EB2 low pt
0045     cutCategory2 = "-1830.8583661119892 - exp(-pt /   -36578.11055382301) * -1831.2083578116517", # EE low pt
0046     cutCategory3 = "0.9717674837607253 - exp(-pt /    8.912850985100356) *  1.9712414940437244", # EB1
0047     cutCategory4 = "0.9458745023265976 - exp(-pt /     8.83104420392795) *    2.40849932040698", # EB2
0048     cutCategory5 = "0.8979112012086751 - exp(-pt /    9.814082144168015) *   4.171581694893849", # EE
0049 )
0050
0051 idName80 = "mvaEleID-Fall17-iso-V1-wp80"
0052 MVA_WP80 = EleMVA_WP(
0053     idName80, mvaTag,
0054     cutCategory0 = "0.9725509559754997 - exp(-pt /  2.976593261509491) *  0.2653858736397496", # EB1 low pt
0055     cutCategory1 = "0.9508038141601247 - exp(-pt / 2.6633500558725713) *  0.2355820499260076", # EB2 low pt
0056     cutCategory2 = "0.9365037167596238 - exp(-pt / 1.5765442323949856) *   3.067015289215309", # EE low pt
0057     cutCategory3 = "0.9896562087723659 - exp(-pt / 10.342490511998674) * 0.40204156417414094", # EB1
0058     cutCategory4 = "0.9819232656533827 - exp(-pt /   9.05548836482051) *   0.772674931169389", # EB2
0059     cutCategory5 = "0.9625098201744635 - exp(-pt /   8.42589315557279) *  2.2916152615134173", # EE
0060 )
0061
0062 ### WP tuned for HZZ analysis with very high efficiency (about 98%)
0063 # The working points were found by requiring the same signal efficiencies in
0064 # each category as for the Spring 16 HZZ ID
0065 # (see RecoEgamma/ElectronIdentification/python/Identification/mvaElectronID_Spring16_HZZ_V1_cff.py)
0066 idNamewpLoose = "mvaEleID-Fall17-iso-V1-wpLoose"
0067 MVA_WPLoose = EleMVA_WP(
0068     idNamewpLoose, mvaTag,
0069     cutCategory0 =  "-0.09564086146419018", # EB1 low pt
0070     cutCategory1 =  "-0.28229916981926795", # EB2 low pt
0071     cutCategory2 =  "-0.05466682296962322", # EE low pt
0072     cutCategory3 =  "-0.833466688584422"  , # EB1
0073     cutCategory4 =  "-0.7677000247570116" , # EB2
0074     cutCategory5 =  "-0.6917305995653829"   # EE
0075     )
0076
0077 #
0078 # Finally, set up VID configuration for all cuts
0079 #
0080
0081 # Create the PSet that will be fed to the MVA value map producer
0082 mvaEleID_Fall17_iso_V1_producer_config = cms.PSet(
0083     mvaName             = cms.string(mvaClassName),
0084     mvaTag              = cms.string(mvaTag),
0085     # Category parameters
0086     nCategories         = cms.int32(6),
0087     categoryCuts        = cms.vstring(*EleMVA_6CategoriesCuts),
0088     # Weight files and variable definitions
0089     weightFileNames     = mvaFall17WeightFiles_V1,
0090     variableDefinition  = cms.string("RecoEgamma/ElectronIdentification/data/ElectronMVAEstimatorRun2Fall17V1Variables.txt")
0091     )
0092 # Create the VPset's for VID cuts
0093 mvaEleID_Fall17_V1_wpLoose = configureVIDMVAEleID( MVA_WPLoose )
0094 mvaEleID_Fall17_V1_wp90 = configureVIDMVAEleID( MVA_WP90)
0095 mvaEleID_Fall17_V1_wp80 = configureVIDMVAEleID( MVA_WP80)
0096
0097 mvaEleID_Fall17_V1_wpLoose.isPOGApproved = cms.untracked.bool(True)
0098 mvaEleID_Fall17_V1_wp90.isPOGApproved = cms.untracked.bool(True)
0099 mvaEleID_Fall17_V1_wp80.isPOGApproved = cms.untracked.bool(True)