Simply preprocess params.

cms-sw · Mar 3, 2020 · d1af63d · d1af63d
1 parent 4d8bfcb
commit d1af63d
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 248 deletions.
diff --git a/...et/python/Parameters/ParticleNet/V00/pfMassDecorrelatedParticleNetPreprocessParams_cfi.py b/...et/python/Parameters/ParticleNet/V00/pfMassDecorrelatedParticleNetPreprocessParams_cfi.py
diff --git a/RecoBTag/MXNet/python/Parameters/ParticleNet/V00/pfParticleNetPreprocessParams_cfi.py b/RecoBTag/MXNet/python/Parameters/ParticleNet/V00/pfParticleNetPreprocessParams_cfi.py
@@ -269,3 +269,31 @@
         )
     )
 )
+
+pfMassDecorrelatedParticleNetPreprocessParams = pfParticleNetPreprocessParams.clone(
+    pf_features = pfParticleNetPreprocessParams.pf_features.clone(
+        input_shape = [1, 20, 100],
+        var_names = [
+            'pfcand_pt_log_nopuppi',
+            'pfcand_e_log_nopuppi',
+            'pfcand_etarel',
+            'pfcand_phirel',
+            'pfcand_abseta',
+            'pfcand_charge',
+            'pfcand_VTX_ass',
+            'pfcand_lostInnerHits',
+            'pfcand_normchi2',
+            'pfcand_quality',
+            'pfcand_dz',
+            'pfcand_dzsig',
+            'pfcand_dxy',
+            'pfcand_dxysig',
+            'pfcand_btagEtaRel',
+            'pfcand_btagPtRatio',
+            'pfcand_btagPParRatio',
+            'pfcand_btagSip3dVal',
+            'pfcand_btagSip3dSig',
+            'pfcand_btagJetDistVal'
+        ]
+    )
+)
diff --git a/RecoBTag/MXNet/python/pfParticleNet_cff.py b/RecoBTag/MXNet/python/pfParticleNet_cff.py
@@ -2,8 +2,7 @@
 
 from RecoBTag.FeatureTools.pfParticleNetTagInfos_cfi import pfParticleNetTagInfos
 from RecoBTag.MXNet.boostedJetMXNetJetTagsProducer_cfi import boostedJetMXNetJetTagsProducer
-from RecoBTag.MXNet.Parameters.ParticleNet.V00.pfParticleNetPreprocessParams_cfi import pfParticleNetPreprocessParams
-from RecoBTag.MXNet.Parameters.ParticleNet.V00.pfMassDecorrelatedParticleNetPreprocessParams_cfi import pfMassDecorrelatedParticleNetPreprocessParams
+from RecoBTag.MXNet.Parameters.ParticleNet.V00.pfParticleNetPreprocessParams_cfi import pfParticleNetPreprocessParams, pfMassDecorrelatedParticleNetPreprocessParams
 from RecoBTag.MXNet.pfParticleNetDiscriminatorsJetTags_cfi import pfParticleNetDiscriminatorsJetTags
 from RecoBTag.MXNet.pfMassDecorrelatedParticleNetDiscriminatorsJetTags_cfi import pfMassDecorrelatedParticleNetDiscriminatorsJetTags