From 35642ba47cb675db58611ef393d48e541619acb8 Mon Sep 17 00:00:00 2001 From: rkansal47 Date: Wed, 3 Apr 2024 20:27:28 -0500 Subject: [PATCH] vbf vars arg --- src/HHbbVV/postprocessing/TrainBDT.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/HHbbVV/postprocessing/TrainBDT.py b/src/HHbbVV/postprocessing/TrainBDT.py index be26a636..8f8317a1 100644 --- a/src/HHbbVV/postprocessing/TrainBDT.py +++ b/src/HHbbVV/postprocessing/TrainBDT.py @@ -266,6 +266,8 @@ def main(args): label_encoder.classes_ = np.array(training_keys) # need this to maintain training keys order bdtVars = AllTaggerBDTVars if args.all_tagger_vars else SingleTaggerBDTVars + if not args.vbf_vars: + bdtVars = bdtVars[:-4] early_stopping_callback = xgb.callback.EarlyStopping( rounds=args.early_stopping_rounds, min_delta=args.early_stopping_min_delta @@ -818,6 +820,7 @@ def do_inference( add_bool_arg( parser, "all-tagger-vars", "Use all tagger outputs vs. single THWWvsT score", default=True ) + add_bool_arg(parser, "vbf-vars", "Use VBF vars", default=True) add_bool_arg(parser, "multiclass", "Classify each background separately", default=True) add_bool_arg(parser, "use-sample-weights", "Use properly scaled event weights", default=True)