CTagging/python/charmTagsComputerCvsL_cfi.py

0001 import FWCore.ParameterSet.Config as cms
0002 #use import as to mask it to process.load()
0003 import RecoBTag.SecondaryVertex.candidateCombinedSecondaryVertexSoftLeptonComputer_cfi as sl_cfg
0004 from RecoBTag.CTagging.training_settings import c_vs_l_vars_vpset, c_vs_b_vars_vpset
0005 #from RecoBTag.SecondaryVertex.combinedSecondaryVertexCommon_cff import *
0006
0007 #
0008 # Normal tagger
0009 #
0010
0011 charmTagsComputerCvsL = cms.ESProducer(
0012     'CharmTaggerESProducer',
0013     #clone the cfg only
0014     slComputerCfg = cms.PSet(
0015         **sl_cfg.candidateCombinedSecondaryVertexSoftLeptonComputer.parameters_()
0016     ),
0017     weightFile = cms.FileInPath('RecoBTag/CTagging/data/c_vs_udsg_sklearn.weight.xml'),
0018     variables = c_vs_l_vars_vpset,
0019     mvaName = cms.string('BDT'),
0020     useCondDB = cms.bool(False),
0021     gbrForestLabel = cms.string(''),
0022     useGBRForest = cms.bool(True),
0023     useAdaBoost = cms.bool(False),
0024     defaultValueNoTracks = cms.bool(False)
0025 )
0026
0027 charmTagsComputerCvsL.slComputerCfg.correctVertexMass = False
0028
0029 from Configuration.Eras.Modifier_phase1Pixel_cff import phase1Pixel
0030 phase1Pixel.toModify(charmTagsComputerCvsL,
0031                      weightFile = 'RecoBTag/CTagging/data/c_vs_udsg_PhaseI.xml',
0032                      slComputerCfg = dict(correctVertexMass = True),
0033                      defaultValueNoTracks = True
0034 )