SoftLepton/test/writeGBRForests_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 process = cms.Process("writeGBRForests")
0004
0005 process.maxEvents = cms.untracked.PSet(
0006     input = cms.untracked.int32(1) # NB: needs to be set to 1 so that GBRForestWriter::analyze method gets called exactly once
0007 )
0008
0009 process.source = cms.Source("EmptySource")
0010
0011 process.load('Configuration/StandardSequences/Services_cff')
0012
0013 process.gbrForestWriter = cms.EDAnalyzer("GBRForestWriter",
0014     jobs = cms.VPSet(
0015         cms.PSet(
0016             inputFileName = cms.FileInPath('RecoBTag/SoftLepton/data/SoftPFElectron_BDT.weights.xml.gz'),
0017             inputFileType = cms.string("XML"),
0018             inputVariables = cms.vstring("sip3d", "sip2d", "ptRel", "deltaR", "ratio", "mva_e_pi"),
0019             spectatorVariables = cms.vstring(),
0020             methodName = cms.string("BDT"),
0021             outputFileType = cms.string("SQLLite"),
0022             outputRecord = cms.string("btag_SoftPFElectron_BDT_TMVAv420_74X_v1")
0023         ),
0024         cms.PSet(
0025             inputFileName = cms.FileInPath('RecoBTag/SoftLepton/data/SoftPFMuon_BDT.weights.xml.gz'),
0026             inputFileType = cms.string("XML"),
0027             inputVariables = cms.vstring("TagInfo1.sip3d", "TagInfo1.sip2d", "TagInfo1.ptRel", "TagInfo1.deltaR", "TagInfo1.ratio"),
0028             spectatorVariables = cms.vstring(),
0029             methodName = cms.string("BDT"),
0030             outputFileType = cms.string("SQLLite"),
0031             outputRecord = cms.string("btag_SoftPFMuon_BDT_TMVAv420_74X_v1")
0032         )
0033     )
0034 )
0035
0036 process.load("CondCore.DBCommon.CondDBCommon_cfi")
0037 process.CondDBCommon.connect = 'sqlite_file:btag_SoftPFLepton_BDT_TMVAv420_GBRForest_74X_v1.db'
0038
0039 process.PoolDBOutputService = cms.Service("PoolDBOutputService",
0040     process.CondDBCommon,
0041     timetype = cms.untracked.string('runnumber'),
0042     toPut = cms.VPSet(
0043         cms.PSet(
0044             record = cms.string('btag_SoftPFElectron_BDT_TMVAv420_74X_v1'),
0045             tag = cms.string('btag_SoftPFElectron_BDT_TMVAv420_74X_v1'),
0046             label = cms.untracked.string('btag_SoftPFElectron_BDT')
0047         ),
0048         cms.PSet(
0049             record = cms.string('btag_SoftPFMuon_BDT_TMVAv420_74X_v1'),
0050             tag = cms.string('btag_SoftPFMuon_BDT_TMVAv420_74X_v1'),
0051             label = cms.untracked.string('btag_SoftPFMuon_BDT')
0052         )
0053     )
0054 )
0055
0056 process.p = cms.Path(process.gbrForestWriter)