File indexing completed on 2024-11-25 02:29:17
0001 import FWCore.ParameterSet.Config as cms
0002
0003 def MVATrainingNtuple(*args, **kwargs):
0004 mod = cms.EDAnalyzer('MVATrainingNtuple',
0005 inputTracks = cms.InputTag('generalTracks'),
0006 inputTagT = cms.InputTag('trackExtenderWithMTD'),
0007 inputTagV = cms.InputTag('offlinePrimaryVertices4D'),
0008 TPtoRecoTrackAssoc = cms.InputTag('trackingParticleRecoTrackAsssociation'),
0009 tp2SimAssociationMapTag = cms.InputTag('mtdSimLayerClusterToTPAssociation'),
0010 r2sAssociationMapTag = cms.InputTag('mtdRecoClusterToSimLayerClusterAssociation'),
0011 SimTag = cms.InputTag('mix', 'MergedTrackTruth'),
0012 offlineBS = cms.InputTag('offlineBeamSpot'),
0013 trackAssocSrc = cms.InputTag('trackExtenderWithMTD', 'generalTrackassoc'),
0014 recCluTagBTL = cms.InputTag('mtdClusters', 'FTLBarrel'),
0015 recCluTagETL = cms.InputTag('mtdClusters', 'FTLEndcap'),
0016 pathLengthSrc = cms.InputTag('trackExtenderWithMTD', 'generalTrackPathLength'),
0017 momentumSrc = cms.InputTag('trackExtenderWithMTD', 'generalTrackp'),
0018 tmtd = cms.InputTag('trackExtenderWithMTD', 'generalTracktmtd'),
0019 sigmaSrc = cms.InputTag('trackExtenderWithMTD', 'generalTracksigmatmtd'),
0020 t0Src = cms.InputTag('trackExtenderWithMTD', 'generalTrackt0'),
0021 sigmat0Src = cms.InputTag('trackExtenderWithMTD', 'generalTracksigmat0'),
0022 t0PID = cms.InputTag('tofPID', 't0'),
0023 sigmat0PID = cms.InputTag('tofPID', 'sigmat0'),
0024 t0SafePID = cms.InputTag('tofPID', 't0safe'),
0025 sigmat0SafePID = cms.InputTag('tofPID', 'sigmat0safe'),
0026 trackMVAQual = cms.InputTag('mtdTrackQualityMVA', 'mtdQualMVA'),
0027 tofPi = cms.InputTag('trackExtenderWithMTD', 'generalTrackTofPi'),
0028 tofK = cms.InputTag('trackExtenderWithMTD', 'generalTrackTofK'),
0029 tofP = cms.InputTag('trackExtenderWithMTD', 'generalTrackTofP'),
0030 probPi = cms.InputTag('tofPID', 'probPi'),
0031 probK = cms.InputTag('tofPID', 'probK'),
0032 probP = cms.InputTag('tofPID', 'probP'),
0033 sigmatofpiSrc = cms.InputTag('trackExtenderWithMTD', 'generalTrackSigmaTofPi'),
0034 sigmatofkSrc = cms.InputTag('trackExtenderWithMTD', 'generalTrackSigmaTofK'),
0035 sigmatofpSrc = cms.InputTag('trackExtenderWithMTD', 'generalTrackSigmaTofP'),
0036 btlMatchChi2Src = cms.InputTag('trackExtenderWithMTD', 'btlMatchChi2'),
0037 btlMatchTimeChi2Src = cms.InputTag('trackExtenderWithMTD', 'btlMatchTimeChi2'),
0038 etlMatchChi2Src = cms.InputTag('trackExtenderWithMTD', 'etlMatchChi2'),
0039 etlMatchTimeChi2Src = cms.InputTag('trackExtenderWithMTD', 'etlMatchTimeChi2'),
0040 npixBarrelSrc = cms.InputTag('trackExtenderWithMTD', 'npixBarrel'),
0041 npixEndcapSrc = cms.InputTag('trackExtenderWithMTD', 'npixEndcap'),
0042 outermostHitPositionSrc = cms.InputTag('trackExtenderWithMTD', 'generalTrackOutermostHitPosition'),
0043 fileName = cms.untracked.string('file.root'),
0044 ntupleforBDT = cms.bool(False),
0045 ntupleforGNN = cms.bool(True),
0046 TkFilterParameters = cms.PSet(
0047 maxNormalizedChi2 = cms.double(10),
0048 minPt = cms.double(0),
0049 algorithm = cms.string('filter'),
0050 maxEta = cms.double(2.4),
0051 maxD0Significance = cms.double(4),
0052 maxD0Error = cms.double(1),
0053 maxDzError = cms.double(1),
0054 trackQuality = cms.string('any'),
0055 minPixelLayersWithHits = cms.int32(2),
0056 minSiliconLayersWithHits = cms.int32(5),
0057 minValidStripHits = cms.int32(0),
0058 numTracksThreshold = cms.int32(0),
0059 maxNumTracksThreshold = cms.int32(2147483647),
0060 minPtTight = cms.double(0)
0061 ),
0062 mightGet = cms.optional.untracked.vstring
0063 )
0064 for a in args:
0065 mod.update_(a)
0066 mod.update_(kwargs)
0067 return mod
