Combined/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/Combined/data/CombinedMVAV2_13_07_2015.weights.xml.gz'),
0017             inputFileType = cms.string("XML"),
0018             inputVariables = cms.vstring("Jet_CSV", "Jet_CSVIVF", "Jet_JP", "Jet_JBP", "Jet_SoftMu", "Jet_SoftEl"),
0019             spectatorVariables = cms.vstring(),
0020             methodName = cms.string("BDT"),
0021             outputFileType = cms.string("SQLLite"),
0022             outputRecord = cms.string("btag_CombinedMVAv2_BDT_TMVAv420_74X_v1")
0023         )
0024     )
0025 )
0026
0027 process.load("CondCore.DBCommon.CondDBCommon_cfi")
0028 process.CondDBCommon.connect = 'sqlite_file:btag_CombinedMVAv2_BDT_TMVAv420_GBRForest_74X_v1.db'
0029
0030 process.PoolDBOutputService = cms.Service("PoolDBOutputService",
0031     process.CondDBCommon,
0032     timetype = cms.untracked.string('runnumber'),
0033     toPut = cms.VPSet(
0034         cms.PSet(
0035             record = cms.string('btag_CombinedMVAv2_BDT_TMVAv420_74X_v1'),
0036             tag = cms.string('btag_CombinedMVAv2_BDT_TMVAv420_74X_v1'),
0037             label = cms.untracked.string('btag_CombinedMVAv2_BDT')
0038         )
0039     )
0040 )
0041
0042 process.p = cms.Path(process.gbrForestWriter)