From 5139caccee12f58651a5f704350044fed7bcd5c7 Mon Sep 17 00:00:00 2001 From: rkansal47 Date: Fri, 3 Nov 2023 14:07:46 -0500 Subject: [PATCH] bf --- src/HHbbVV/postprocessing/TrainBDT.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/HHbbVV/postprocessing/TrainBDT.py b/src/HHbbVV/postprocessing/TrainBDT.py index acdd76ee..4c4b5b8f 100644 --- a/src/HHbbVV/postprocessing/TrainBDT.py +++ b/src/HHbbVV/postprocessing/TrainBDT.py @@ -545,7 +545,7 @@ def evaluate_model( for dataset, label in [(train, "Train"), (test, "Test")]: # Normalize the two distributions - sf = 0.5 / test_size if label == "Test" else (1 - test_size) / 0.5 + sf = (0.5 / test_size) if label == "Test" else (0.5 / (1 - test_size)) print(f"Scaling {label} by {sf}") for key in training_keys: data = dataset[year][dataset[year]["Dataset"] == key]