/
charmTaggerProducer_cfi.py
34 lines (30 loc) · 1.3 KB
/
charmTaggerProducer_cfi.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import FWCore.ParameterSet.Config as cms
#use import as to mask it to process.load()
import RecoBTag.SecondaryVertex.candidateCombinedSecondaryVertexSoftLeptonComputer_cfi as sl_cfg
from RecoBTag.CTagging.training_settings import c_vs_l_vars_vpset, c_vs_b_vars_vpset
charmTagsComputerCvsL = cms.ESProducer(
'CharmTaggerESProducer',
#clone the cfg only
slComputerCfg = cms.PSet(
**sl_cfg.candidateCombinedSecondaryVertexSoftLeptonComputer.parameters_()
),
weightFile = cms.FileInPath('RecoBTag/CTagging/data/c_vs_udsg_sklearn.weight.xml'),
variables = c_vs_l_vars_vpset,
computer = cms.ESInputTag('combinedSecondaryVertexSoftLeptonComputer'),
tagInfos = cms.VInputTag(
cms.InputTag('pfImpactParameterTagInfos'),
cms.InputTag('pfInclusiveSecondaryVertexFinderCvsLTagInfos'),
cms.InputTag('softPFMuonsTagInfos'),
cms.InputTag('softPFElectronsTagInfos'),
),
mvaName = cms.string('BDT'),
useCondDB = cms.bool(False),
gbrForestLabel = cms.string(''),
useGBRForest = cms.bool(True),
useAdaBoost = cms.bool(False)
)
charmTagsComputerCvsL.slComputerCfg.correctVertexMass = False
charmTagsComputerCvsB = charmTagsComputerCvsL.clone(
weightFile = cms.FileInPath('RecoBTag/CTagging/data/c_vs_b_sklearn.weight.xml'),
variables = c_vs_b_vars_vpset
)