Merge branch 'bdt_disc' of github.com:LPC-HH/HH4b into bdt_disc

LPC-HH · Nov 15, 2024 · de7d099 · de7d099
2 parents 23766ba + abc6435
commit de7d099
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/HH4b/processors/objects.py b/src/HH4b/processors/objects.py
@@ -263,8 +263,8 @@ def get_ak8jets(fatjets: FatJetArray):
         # T for discriminator
         fatjets["ParTTXbb"] = fatjets.globalParT_XbbVsQCD
         # Mass Regression
-        fatjets["ParTmassRes"] = fatjets.globalParT_massRes * fatjets.mass
-        fatjets["ParTmassVis"] = fatjets.globalParT_massVis * fatjets.mass
+        fatjets["ParTmassRes"] = fatjets.globalParT_massRes * (1 - fatjets.rawFactor) * fatjets.mass
+        fatjets["ParTmassVis"] = fatjets.globalParT_massVis * (1 - fatjets.rawFactor) * fatjets.mass
 
     return fatjets