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    computer = cms.ESInputTag('combinedSecondaryVertexSoftLeptonComputer',''),
0020    tagInfos = cms.VInputTag(
0021       cms.InputTag('pfImpactParameterTagInfos'),
0022       cms.InputTag('pfInclusiveSecondaryVertexFinderCvsLTagInfos'),
0023       cms.InputTag('softPFMuonsTagInfos'),
0024       cms.InputTag('softPFElectronsTagInfos'),
0025       ),
0026    mvaName = cms.string('BDT'),
0027    useCondDB = cms.bool(False),
0028    gbrForestLabel = cms.string(''),
0029    useGBRForest = cms.bool(True),
0030    useAdaBoost = cms.bool(False),
0031    defaultValueNoTracks = cms.bool(False)
0032    )
0033
0034 charmTagsComputerCvsL.slComputerCfg.correctVertexMass = False
0035
0036 from Configuration.Eras.Modifier_phase1Pixel_cff import phase1Pixel
0037 phase1Pixel.toModify(charmTagsComputerCvsL,
0038                      weightFile = 'RecoBTag/CTagging/data/c_vs_udsg_PhaseI.xml',
0039                      slComputerCfg = dict(correctVertexMass = True),
0040                      defaultValueNoTracks = True
0041 )