00001 import FWCore.ParameterSet.Config as cms
00002
00003 """
00004 TauMVAConfigurations_cfi.py
00005 Author: Evan K. Friis, UC Davis (friis@physics.ucdavis.edu)
00006
00007 Define names, locations, and decay mode mappings of different MVA configurations
00008
00009 At the bottom of this file the collections of individual neural nets are defined (eg TaNC).
00010 These collections are the inputs to the actual discriminator
00011
00012 Example:
00013
00014 OneProngs = cms.PSet( # name
00015 decayModeIndices = cms.vint32(0, 1, 2, 3, 4), # Decay modes this MVA applies to. In this case, decay modes with 1 track and 0, 1, 2, 3, 4 pi zeros.
00016 # Please see DataFormats/TauReco/interface/PFTauDecayMode.h
00017 # for the definition of the decay mode indexing system. [DecayMode# = (#ChargedPions-1) * 5 + #PiZeros]
00018 computerName = cms.string('OneProngs') # Identifier string. Will be used to store the trainined MVA data in the database
00019 # When training, the corresponding training xml file must be in RecoTauTag/TauTagTools/xml/[computerName].xml (i.e. OneProngs.xml)
00020 applyIsolation = cms.bool(False), # Determines if isolation criteria is applied (for both training and final discrimination)
00021
00022 cut = cms.double(-10.) # Serves as a place holder for when the user wishes to specify the cuts they wish to use
00023 )
00024
00025 """
00026
00027 OneProngNoPiZero = cms.PSet(
00028 decayModeIndices = cms.vint32(0),
00029 computerName = cms.string('OneProngNoPiZero'),
00030 applyIsolation = cms.bool(False),
00031 cut = cms.double(-10.)
00032 )
00033
00034 OneProngOnePiZero = cms.PSet(
00035 decayModeIndices = cms.vint32(1),
00036 computerName = cms.string('OneProngOnePiZero'),
00037 applyIsolation = cms.bool(False),
00038 cut = cms.double(-10.)
00039 )
00040
00041 OneProngTwoPiZero = cms.PSet(
00042 decayModeIndices = cms.vint32(2),
00043 computerName = cms.string('OneProngTwoPiZero'),
00044 applyIsolation = cms.bool(False),
00045 cut = cms.double(-10.)
00046 )
00047
00048 ThreeProngNoPiZero = cms.PSet(
00049 decayModeIndices = cms.vint32(10),
00050 computerName = cms.string('ThreeProngNoPiZero'),
00051 applyIsolation = cms.bool(False),
00052 cut = cms.double(-10.)
00053 )
00054
00055 ThreeProngOnePiZero = cms.PSet(
00056 decayModeIndices = cms.vint32(11),
00057 computerName = cms.string('ThreeProngOnePiZero'),
00058 applyIsolation = cms.bool(False),
00059 cut = cms.double(-10.)
00060 )
00061
00062 SingleNet = cms.PSet(
00063 decayModeIndices = cms.vint32(0, 1, 2, 10, 11),
00064 computerName = cms.string('SingleNet'),
00065 applyIsolation = cms.bool(False),
00066 cut = cms.double(-10.)
00067 )
00068
00069
00070
00071 OneProngNoPiZeroIso = cms.PSet(
00072 decayModeIndices = cms.vint32(0),
00073 computerName = cms.string('OneProngNoPiZeroIso'),
00074 applyIsolation = cms.bool(True),
00075 cut = cms.double(-10.)
00076 )
00077
00078 OneProngOnePiZeroIso = cms.PSet(
00079 decayModeIndices = cms.vint32(1),
00080 computerName = cms.string('OneProngOnePiZeroIso'),
00081 applyIsolation = cms.bool(True),
00082 cut = cms.double(-10.)
00083 )
00084
00085 OneProngTwoPiZeroIso = cms.PSet(
00086 decayModeIndices = cms.vint32(2),
00087 computerName = cms.string('OneProngTwoPiZeroIso'),
00088 applyIsolation = cms.bool(True),
00089 cut = cms.double(-10.)
00090 )
00091
00092 ThreeProngNoPiZeroIso = cms.PSet(
00093 decayModeIndices = cms.vint32(10),
00094 computerName = cms.string('ThreeProngNoPiZeroIso'),
00095 applyIsolation = cms.bool(True),
00096 cut = cms.double(-10.)
00097 )
00098
00099 ThreeProngOnePiZeroIso = cms.PSet(
00100 decayModeIndices = cms.vint32(11),
00101 computerName = cms.string('ThreeProngOnePiZeroIso'),
00102 applyIsolation = cms.bool(True),
00103 cut = cms.double(-10.)
00104 )
00105
00106 SingleNetIso = cms.PSet(
00107 decayModeIndices = cms.vint32(0, 1, 2, 10, 11),
00108 computerName = cms.string('SingleNetIso'),
00109 applyIsolation = cms.bool(True),
00110 cut = cms.double(-10.)
00111 )
00112
00113
00114
00115
00116 TaNC = cms.VPSet(
00117 OneProngNoPiZero,
00118 OneProngOnePiZero,
00119 OneProngTwoPiZero,
00120 ThreeProngNoPiZero,
00121 ThreeProngOnePiZero
00122 )
00123
00124 MultiNetIso = cms.VPSet(
00125 OneProngNoPiZeroIso,
00126 OneProngOnePiZeroIso,
00127 OneProngTwoPiZeroIso,
00128 ThreeProngNoPiZeroIso,
00129 ThreeProngOnePiZeroIso
00130 )
00131
00132 SingleNetBasedTauID = cms.VPSet(
00133 SingleNet
00134 )