TrackTrigger/python/TrackQualityParams_cfi.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 TrackQualityParams = cms.PSet(qualityAlgorithm = cms.string("GBDT"), #None, Cut, NN, GBDT
0004                               ONNXmodel = cms.FileInPath("L1Trigger/TrackTrigger/data/GBDT_default.onnx"),
0005                               # The ONNX model should be found at this path, if you want a local version of the model:
0006                               # git clone https://github.com/cms-data/L1Trigger-TrackTrigger.git L1Trigger/TrackTrigger/data
0007                               ONNXInputName = cms.string("feature_input"),
0008                               #Vector of strings of training features, in the order that the model was trained with
0009                               featureNames = cms.vstring(["phi", "eta", "z0", "bendchi2_bin", "nstub",
0010                                                           "nlaymiss_interior", "chi2rphi_bin", "chi2rz_bin"]),
0011                               # Parameters for cut based classifier, optimized for L1 Track MET
0012                               # (Table 3.7  The Phase-2 Upgrade of the CMS Level-1 Trigger http://cds.cern.ch/record/2714892)
0013                               maxZ0 = cms.double ( 15. ) ,    # in cm
0014                               maxEta = cms.double ( 2.4 ) ,
0015                               chi2dofMax = cms.double( 40. ),
0016                               bendchi2Max = cms.double( 2.4 ),
0017                               minPt = cms.double( 2. ),       # in GeV
0018                               nStubsmin = cms.int32( 4 ),
0019                               )