Skip to content

Commit

Permalink
Keep DeepBoostedJet V01 as default.
Browse files Browse the repository at this point in the history
  • Loading branch information
@hqucms
hqucms committed Apr 14, 2019
1 parent 423c632 commit 5eb90b9
Show file tree
Hide file tree
Showing 4 changed files with 329 additions and 6 deletions.
309 changes: 309 additions & 0 deletions RecoBTag/MXNet/python/Parameters/V01/pfDeepBoostedJetPreprocessParams_cfi.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,309 @@
import FWCore.ParameterSet.Config as cms

pfDeepBoostedJetPreprocessParams = cms.PSet(
input_names = cms.vstring(
'pfcand',
'sv'
),
pfcand = cms.PSet(
input_shape = cms.vuint32(1, 42, 100, 1),
var_infos = cms.PSet(
pfcand_VTX_ass = cms.PSet(
median = cms.double(7.0),
norm_factor = cms.double(1.0)
),
pfcand_abseta = cms.PSet(
median = cms.double(0.599505603313),
norm_factor = cms.double(1.62486364812)
),
pfcand_btagEtaRel = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(0.30955120799)
),
pfcand_btagJetDistVal = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_btagPParRatio = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.00271135589)
),
pfcand_btagPtRatio = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(5.73484698484)
),
pfcand_btagSip2dSig = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.76905343449)
),
pfcand_btagSip2dVal = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(605.021135542)
),
pfcand_btagSip3dSig = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(0.9666446673)
),
pfcand_btagSip3dVal = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(239.708840806)
),
pfcand_charge = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_deltaR = cms.PSet(
median = cms.double(0.22575956583),
norm_factor = cms.double(3.81050649022)
),
pfcand_detadeta = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1967329.39697)
),
pfcand_dlambdadz = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_dphidphi = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1310150.11117)
),
pfcand_dphidxy = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_dptdpt = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(163433.258365)
),
pfcand_drminsv = cms.PSet(
median = cms.double(0.177426457405),
norm_factor = cms.double(2.64722103751)
),
pfcand_drsubjet1 = cms.PSet(
median = cms.double(0.231124095619),
norm_factor = cms.double(3.14072264308)
),
pfcand_drsubjet2 = cms.PSet(
median = cms.double(0.263272643089),
norm_factor = cms.double(2.922280097)
),
pfcand_dxy = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(392.713315451)
),
pfcand_dxydxy = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(46519.023087)
),
pfcand_dxydz = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(7626007.10744)
),
pfcand_dxysig = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(2.01382982135)
),
pfcand_dz = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(222.850935094)
),
pfcand_dzdz = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(29383.0691756)
),
pfcand_dzsig = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.67618760674)
),
pfcand_erel_log = cms.PSet(
median = cms.double(-5.38983869553),
norm_factor = cms.double(0.539102612572)
),
pfcand_etarel = cms.PSet(
median = cms.double(-0.0054658302106),
norm_factor = cms.double(5.54556174395)
),
pfcand_hcalFrac = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_isChargedHad = cms.PSet(
median = cms.double(1.0),
norm_factor = cms.double(1.0)
),
pfcand_isEl = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_isGamma = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_isMu = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_isNeutralHad = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(1.0)
),
pfcand_lostInnerHits = cms.PSet(
median = cms.double(-1.0),
norm_factor = cms.double(1.0)
),
pfcand_normchi2 = cms.PSet(
median = cms.double(999.0),
norm_factor = cms.double(1.0)
),
pfcand_phirel = cms.PSet(
median = cms.double(-5.10289683007e-05),
norm_factor = cms.double(4.63705896375)
),
pfcand_pt_log = cms.PSet(
median = cms.double(1.09469842911),
norm_factor = cms.double(0.518874132164)
),
pfcand_ptrel_log = cms.PSet(
median = cms.double(-5.38205528259),
norm_factor = cms.double(0.537919766379)
),
pfcand_puppiw = cms.PSet(
median = cms.double(1.0),
norm_factor = cms.double(1.0)
),
pfcand_quality = cms.PSet(
median = cms.double(0.0),
norm_factor = cms.double(0.2)
)
),
var_length = cms.uint32(100),
var_names = cms.vstring(
'pfcand_pt_log',
'pfcand_ptrel_log',
'pfcand_erel_log',
'pfcand_phirel',
'pfcand_etarel',
'pfcand_deltaR',
'pfcand_abseta',
'pfcand_puppiw',
'pfcand_drminsv',
'pfcand_drsubjet1',
'pfcand_drsubjet2',
'pfcand_charge',
'pfcand_isMu',
'pfcand_isEl',
'pfcand_isChargedHad',
'pfcand_isGamma',
'pfcand_isNeutralHad',
'pfcand_hcalFrac',
'pfcand_VTX_ass',
'pfcand_lostInnerHits',
'pfcand_normchi2',
'pfcand_quality',
'pfcand_dz',
'pfcand_dzsig',
'pfcand_dxy',
'pfcand_dxysig',
'pfcand_dptdpt',
'pfcand_detadeta',
'pfcand_dphidphi',
'pfcand_dxydxy',
'pfcand_dzdz',
'pfcand_dxydz',
'pfcand_dphidxy',
'pfcand_dlambdadz',
'pfcand_btagEtaRel',
'pfcand_btagPtRatio',
'pfcand_btagPParRatio',
'pfcand_btagSip2dVal',
'pfcand_btagSip2dSig',
'pfcand_btagSip3dVal',
'pfcand_btagSip3dSig',
'pfcand_btagJetDistVal'
)
),
sv = cms.PSet(
input_shape = cms.vuint32(1, 15, 7, 1),
var_infos = cms.PSet(
sv_abseta = cms.PSet(
median = cms.double(0.579698801041),
norm_factor = cms.double(1.5478785279)
),
sv_costhetasvpv = cms.PSet(
median = cms.double(0.999744534492),
norm_factor = cms.double(4035.89511667)
),
sv_d3d = cms.PSet(
median = cms.double(0.496477723122),
norm_factor = cms.double(0.230671083153)
),
sv_d3dsig = cms.PSet(
median = cms.double(6.50515079498),
norm_factor = cms.double(0.0328928080185)
),
sv_deltaR = cms.PSet(
median = cms.double(0.105607174337),
norm_factor = cms.double(4.58211966501)
),
sv_dxy = cms.PSet(
median = cms.double(0.369336694479),
norm_factor = cms.double(0.2771604275)
),
sv_dxysig = cms.PSet(
median = cms.double(6.49603271484),
norm_factor = cms.double(0.0328965270602)
),
sv_erel_log = cms.PSet(
median = cms.double(-2.77275466919),
norm_factor = cms.double(0.976448084421)
),
sv_etarel = cms.PSet(
median = cms.double(-0.00318800867535),
norm_factor = cms.double(10.1732672714)
),
sv_mass = cms.PSet(
median = cms.double(1.36728298664),
norm_factor = cms.double(0.294783235965)
),
sv_normchi2 = cms.PSet(
median = cms.double(0.796081960201),
norm_factor = cms.double(0.737576078452)
),
sv_ntracks = cms.PSet(
median = cms.double(3.0),
norm_factor = cms.double(0.5)
),
sv_phirel = cms.PSet(
median = cms.double(0.000510219600983),
norm_factor = cms.double(9.11781369578)
),
sv_pt_log = cms.PSet(
median = cms.double(3.77797603607),
norm_factor = cms.double(0.889925735415)
),
sv_ptrel_log = cms.PSet(
median = cms.double(-2.7622461319),
norm_factor = cms.double(0.975634242485)
)
),
var_length = cms.uint32(7),
var_names = cms.vstring(
'sv_pt_log',
'sv_ptrel_log',
'sv_erel_log',
'sv_phirel',
'sv_etarel',
'sv_deltaR',
'sv_abseta',
'sv_mass',
'sv_ntracks',
'sv_normchi2',
'sv_dxy',
'sv_dxysig',
'sv_d3d',
'sv_d3dsig',
'sv_costhetasvpv'
)
)
)
0 ...n/pfDeepBoostedJetPreprocessParams_cfi.py → ...2/pfDeepBoostedJetPreprocessParams_cfi.py
View file
File renamed without changes.
0 ...atedDeepBoostedJetPreprocessParams_cfi.py → ...atedDeepBoostedJetPreprocessParams_cfi.py
View file
File renamed without changes.
26 changes: 20 additions & 6 deletions RecoBTag/MXNet/python/pfDeepBoostedJet_cff.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,25 +2,39 @@

from RecoBTag.FeatureTools.pfDeepBoostedJetTagInfos_cfi import pfDeepBoostedJetTagInfos
from RecoBTag.MXNet.pfDeepBoostedJetTags_cfi import pfDeepBoostedJetTags as _pfDeepBoostedJetTags
from RecoBTag.MXNet.pfDeepBoostedJetPreprocessParams_cfi import pfDeepBoostedJetPreprocessParams
from RecoBTag.MXNet.pfMassDecorrelatedDeepBoostedJetPreprocessParams_cfi import pfMassDecorrelatedDeepBoostedJetPreprocessParams
from RecoBTag.MXNet.Parameters.V01.pfDeepBoostedJetPreprocessParams_cfi import pfDeepBoostedJetPreprocessParams
from RecoBTag.MXNet.pfDeepBoostedDiscriminatorsJetTags_cfi import pfDeepBoostedDiscriminatorsJetTags
from RecoBTag.MXNet.pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags_cfi import pfMassDecorrelatedDeepBoostedDiscriminatorsJetTags

# nominal DeepAK8
pfDeepBoostedJetTags = _pfDeepBoostedJetTags.clone(
preprocessParams = pfDeepBoostedJetPreprocessParams,
model_path = 'RecoBTag/Combined/data/DeepBoostedJet/V02/full/resnet-symbol.json',
param_path = 'RecoBTag/Combined/data/DeepBoostedJet/V02/full/resnet-0000.params',
model_path = 'RecoBTag/Combined/data/DeepBoostedJet/V01/full/resnet-symbol.json',
param_path = 'RecoBTag/Combined/data/DeepBoostedJet/V01/full/resnet-0000.params',
debugMode = False, # debug
)

# mass-decorrelated DeepAK8
pfMassDecorrelatedDeepBoostedJetTags = _pfDeepBoostedJetTags.clone(
preprocessParams = pfMassDecorrelatedDeepBoostedJetPreprocessParams,
preprocessParams = pfDeepBoostedJetPreprocessParams,
model_path = 'RecoBTag/Combined/data/DeepBoostedJet/V01/decorrelated/resnet-symbol.json',
param_path = 'RecoBTag/Combined/data/DeepBoostedJet/V01/decorrelated/resnet-0000.params',
debugMode = False, # debug
)

from Configuration.Eras.Modifier_run2_miniAOD_devel_cff import run2_miniAOD_devel
from RecoBTag.MXNet.Parameters.V02.pfDeepBoostedJetPreprocessParams_cfi import pfDeepBoostedJetPreprocessParams as pfDeepBoostedJetPreprocessParamsV02
from RecoBTag.MXNet.Parameters.V02.pfMassDecorrelatedDeepBoostedJetPreprocessParams_cfi import pfMassDecorrelatedDeepBoostedJetPreprocessParams as pfMassDecorrelatedDeepBoostedJetPreprocessParamsV02

run2_miniAOD_devel.toModify(pfDeepBoostedJetTags,
preprocessParams = pfDeepBoostedJetPreprocessParamsV02,
model_path = 'RecoBTag/Combined/data/DeepBoostedJet/V02/full/resnet-symbol.json',
param_path = 'RecoBTag/Combined/data/DeepBoostedJet/V02/full/resnet-0000.params',
)
run2_miniAOD_devel.toModify(pfMassDecorrelatedDeepBoostedJetTags,
preprocessParams = pfMassDecorrelatedDeepBoostedJetPreprocessParamsV02,
model_path = 'RecoBTag/Combined/data/DeepBoostedJet/V02/decorrelated/resnet-symbol.json',
param_path = 'RecoBTag/Combined/data/DeepBoostedJet/V02/decorrelated/resnet-0000.params',
debugMode = False, # debug
)

from CommonTools.PileupAlgos.Puppi_cff import puppi
Expand Down

0 comments on commit 5eb90b9

Please sign in to comment.